W3cubDocs

/TensorFlow 2.4

Module: tf.compat.v1.tpu

Ops related to Tensor Processing Units.

Modules

experimental module: Public API for tf.tpu.experimental namespace.

Classes

class CrossShardOptimizer: An optimizer that averages gradients across TPU shards.

class PaddingSpec: Represents the type of padding policies for tpu.replicate.

class XLAOptions: XLA compilation options.

Functions

batch_parallel(...): Shards computation along the batch dimension for parallel execution.

bfloat16_scope(...): Scope class for bfloat16 variables so that the model uses custom getter.

core(...): Returns the device name for a core in a replicated TPU computation.

cross_replica_sum(...): Sum the input tensor across replicas according to group_assignment.

initialize_system(...): Initializes a distributed TPU system for use with TensorFlow.

outside_compilation(...): Builds part of a computation outside any current TPU replicate scope.

replicate(...): Builds a graph operator that runs a replicated TPU computation.

rewrite(...): Rewrites computation for execution on a TPU system.

shard(...): Shards computation for parallel execution.

shutdown_system(...): Shuts down a running a distributed TPU system.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/compat/v1/tpu