Public API for tf.estimator.tpu namespace.
Modules
experimental
module: Public API for tf.estimator.tpu.experimental namespace.
Classes
class InputPipelineConfig
: Please see the definition of these values in TPUConfig.
class RunConfig
: RunConfig with TPU support.
class TPUConfig
: TPU related configuration required by TPUEstimator
.
class TPUEstimator
: Estimator with TPU support.
class TPUEstimatorSpec
: Ops and objects returned from a model_fn
and passed to TPUEstimator
.