TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

Module: tf.estimator.tpu

View source on GitHub

Aliases:

  • Module tf.compat.v1.estimator.tpu

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.