Module: tf.estimator.tpu
Stay organized with collections
Save and categorize content based on your preferences.
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
.
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2020-10-01 UTC.
[null,null,["Last updated 2020-10-01 UTC."],[],[],null,["# Module: tf.estimator.tpu\n\n\u003cbr /\u003e\n\nPublic API for tf.estimator.tpu namespace.\n\n#### View aliases\n\n\n**Compat aliases for migration**\n\nSee\n[Migration guide](https://www.tensorflow.org/guide/migrate) for\nmore details.\n\n[`tf.compat.v1.estimator.tpu`](/api_docs/python/tf/compat/v1/estimator/tpu)\n\n\u003cbr /\u003e\n\nModules\n-------\n\n[`experimental`](../../tf/estimator/tpu/experimental) module: Public API for tf.estimator.tpu.experimental namespace.\n\nClasses\n-------\n\n[`class InputPipelineConfig`](../../tf/estimator/tpu/InputPipelineConfig): Please see the definition of these values in TPUConfig.\n\n[`class RunConfig`](../../tf/estimator/tpu/RunConfig): RunConfig with TPU support.\n\n[`class TPUConfig`](../../tf/estimator/tpu/TPUConfig): TPU related configuration required by `TPUEstimator`.\n\n[`class TPUEstimator`](../../tf/estimator/tpu/TPUEstimator): Estimator with TPU support.\n\n[`class TPUEstimatorSpec`](../../tf/estimator/tpu/TPUEstimatorSpec): Ops and objects returned from a `model_fn` and passed to `TPUEstimator`."]]