Type keras_support
Namespace tensorflow.contrib.tpu.python.tpu.keras_support
Methods
Properties
- get_tpu_system_metadata_fn
- KerasCrossShardOptimizer_fn
- KerasTPUModel_fn
- setup_tpu_session_fn
- SizedInfeed_fn
- TPUAssignment_fn
- TPUDatasetInfeedManager_fn
- TPUDistributionStrategy_fn
- TPUEmbedding_fn
- TPUFunction_fn
- TPUInfeedInstance_fn
- TPUInfeedManager_fn
- TPUModelOp_fn
- TPUNumpyInfeedManager_fn
- TPURewriteContext_fn
Fields
Public static methods
object get_tpu_system_metadata(TPUClusterResolver tpu_cluster_resolver)
object get_tpu_system_metadata_dyn(object tpu_cluster_resolver)
void setup_tpu_session(TPUClusterResolver cluster_resolver)
object setup_tpu_session_dyn(object cluster_resolver)
Public properties
PythonFunctionContainer get_tpu_system_metadata_fn get;
PythonFunctionContainer KerasCrossShardOptimizer_fn get;
PythonFunctionContainer KerasTPUModel_fn get;
PythonFunctionContainer setup_tpu_session_fn get;
PythonFunctionContainer SizedInfeed_fn get;
PythonFunctionContainer TPUAssignment_fn get;
PythonFunctionContainer TPUDatasetInfeedManager_fn get;
PythonFunctionContainer TPUDistributionStrategy_fn get;
PythonFunctionContainer TPUEmbedding_fn get;
PythonFunctionContainer TPUFunction_fn get;
PythonFunctionContainer TPUInfeedInstance_fn get;
PythonFunctionContainer TPUInfeedManager_fn get;
PythonFunctionContainer TPUModelOp_fn get;
PythonFunctionContainer TPUNumpyInfeedManager_fn get;
PythonFunctionContainer TPURewriteContext_fn get;
Public fields
object issparse
return object
|