org.tensorflow.op.tpu

Classes

AllToAll <T extends TType > An Op to exchange data across TPU replicas.
CollectivePermute <T extends TType > An Op to permute tensors across replicated TPU instances.
CompilationResult Returns the result of a TPU compilation.
CompileSucceededAssert Asserts that compilation succeeded.
ConfigureDistributedTPU Sets up the centralized structures for a distributed TPU system.
ConfigureDistributedTPU.Options Optional attributes for ConfigureDistributedTPU
ConfigureTPUEmbedding Sets up TPUEmbedding in a distributed TPU system.
CrossReplicaSum <T extends TNumber > An Op to sum inputs across replicated TPU instances.
EmbeddingActivations An op enabling differentiation of TPU Embeddings.
EnqueueTPUEmbeddingIntegerBatch An op that enqueues a list of input batch tensors to TPUEmbedding.
EnqueueTPUEmbeddingIntegerBatch.Options Optional attributes for EnqueueTPUEmbeddingIntegerBatch
EnqueueTPUEmbeddingRaggedTensorBatch Eases the porting of code that uses tf.nn.embedding_lookup().
EnqueueTPUEmbeddingRaggedTensorBatch.Options Optional attributes for EnqueueTPUEmbeddingRaggedTensorBatch
EnqueueTPUEmbeddingSparseBatch An op that enqueues TPUEmbedding input indices from a SparseTensor.
EnqueueTPUEmbeddingSparseBatch.Options Optional attributes for EnqueueTPUEmbeddingSparseBatch
EnqueueTPUEmbeddingSparseTensorBatch Eases the porting of code that uses tf.nn.embedding_lookup_sparse().
EnqueueTPUEmbeddingSparseTensorBatch.Options Optional attributes for EnqueueTPUEmbeddingSparseTensorBatch
Execute Op that loads and executes a TPU program on a TPU device.
ExecuteAndUpdateVariables Op that executes a program with optional in-place variable updates.
InfeedDequeue <T extends TType > A placeholder op for a value that will be fed into the computation.
InfeedDequeueTuple Fetches multiple values from infeed as an XLA tuple.
InfeedEnqueue An op which feeds a single Tensor value into the computation.
InfeedEnqueue.Options Optional attributes for InfeedEnqueue
InfeedEnqueuePrelinearizedBuffer An op which enqueues prelinearized buffer into TPU infeed.
InfeedEnqueuePrelinearizedBuffer.Options Optional attributes for InfeedEnqueuePrelinearizedBuffer
InfeedEnqueueTuple Feeds multiple Tensor values into the computation as an XLA tuple.
InfeedEnqueueTuple.Options Optional attributes for InfeedEnqueueTuple
LoadTPUEmbeddingAdadeltaParameters Load Adadelta embedding parameters.
LoadTPUEmbeddingAdadeltaParameters.Options Optional attributes for LoadTPUEmbeddingAdadeltaParameters
LoadTPUEmbeddingAdadeltaParametersGradAccumDebug Load Adadelta parameters with debug support.
LoadTPUEmbeddingAdadeltaParametersGradAccumDebug.Options Optional attributes for LoadTPUEmbeddingAdadeltaParametersGradAccumDebug
LoadTPUEmbeddingAdagradParameters Load Adagrad embedding parameters.
LoadTPUEmbeddingAdagradParameters.Options Optional attributes for LoadTPUEmbeddingAdagradParameters
LoadTPUEmbeddingAdagradParametersGradAccumDebug Load Adagrad embedding parameters with debug support.
LoadTPUEmbeddingAdagradParametersGradAccumDebug.Options Optional attributes for LoadTPUEmbeddingAdagradParametersGradAccumDebug
LoadTPUEmbeddingADAMParameters Load ADAM embedding parameters.
LoadTPUEmbeddingADAMParameters.Options Optional attributes for LoadTPUEmbeddingADAMParameters
LoadTPUEmbeddingADAMParametersGradAccumDebug Load ADAM embedding parameters with debug support.
LoadTPUEmbeddingADAMParametersGradAccumDebug.Options Optional attributes for LoadTPUEmbeddingADAMParametersGradAccumDebug
LoadTPUEmbeddingCenteredRMSPropParameters Load centered RMSProp embedding parameters.
LoadTPUEmbeddingCenteredRMSPropParameters.Options Optional attributes for LoadTPUEmbeddingCenteredRMSPropParameters
LoadTPUEmbeddingFTRLParameters Load FTRL embedding parameters.
LoadTPUEmbeddingFTRLParameters.Options Optional attributes for LoadTPUEmbeddingFTRLParameters
LoadTPUEmbeddingFTRLParametersGradAccumDebug Load FTRL embedding parameters with debug support.
LoadTPUEmbeddingFTRLParametersGradAccumDebug.Options Optional attributes for LoadTPUEmbeddingFTRLParametersGradAccumDebug
LoadTPUEmbeddingMDLAdagradLightParameters Load MDL Adagrad Light embedding parameters.
LoadTPUEmbeddingMDLAdagradLightParameters.Options Optional attributes for LoadTPUEmbeddingMDLAdagradLightParameters
LoadTPUEmbeddingMomentumParameters Load Momentum embedding parameters.
LoadTPUEmbeddingMomentumParameters.Options Optional attributes for LoadTPUEmbeddingMomentumParameters
LoadTPUEmbeddingMomentumParametersGradAccumDebug Load Momentum embedding parameters with debug support.
LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options Optional attributes for LoadTPUEmbeddingMomentumParametersGradAccumDebug
LoadTPUEmbeddingProximalAdagradParameters Load proximal Adagrad embedding parameters.
LoadTPUEmbeddingProximalAdagradParameters.Options Optional attributes for LoadTPUEmbeddingProximalAdagradParameters
LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug Load proximal Adagrad embedding parameters with debug support.
LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug.Options Optional attributes for LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug
LoadTPUEmbeddingProximalYogiParameters
LoadTPUEmbeddingProximalYogiParameters.Options Optional attributes for LoadTPUEmbeddingProximalYogiParameters
LoadTPUEmbeddingProximalYogiParametersGradAccumDebug
LoadTPUEmbeddingProximalYogiParametersGradAccumDebug.Options Optional attributes for LoadTPUEmbeddingProximalYogiParametersGradAccumDebug
LoadTPUEmbeddingRMSPropParameters Load RMSProp embedding parameters.
LoadTPUEmbeddingRMSPropParameters.Options Optional attributes for LoadTPUEmbeddingRMSPropParameters
LoadTPUEmbeddingRMSPropParametersGradAccumDebug Load RMSProp embedding parameters with debug support.
LoadTPUEmbeddingRMSPropParametersGradAccumDebug.Options Optional attributes for LoadTPUEmbeddingRMSPropParametersGradAccumDebug
LoadTPUEmbeddingStochasticGradientDescentParameters Load SGD embedding parameters.
LoadTPUEmbeddingStochasticGradientDescentParameters.Options Optional attributes for LoadTPUEmbeddingStochasticGradientDescentParameters
LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug Load SGD embedding parameters.
LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.Options Optional attributes for LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug
OrdinalSelector A TPU core selector Op.
OutfeedDequeue <T extends TType > Retrieves a single tensor from the computation outfeed.
OutfeedDequeue.Options Optional attributes for OutfeedDequeue
OutfeedDequeueTuple Retrieve multiple values from the computation outfeed.
OutfeedDequeueTuple.Options Optional attributes for OutfeedDequeueTuple
OutfeedDequeueTupleV2 Retrieve multiple values from the computation outfeed.
OutfeedDequeueV2 <T extends TType > Retrieves a single tensor from the computation outfeed.
OutfeedEnqueue Enqueue a Tensor on the computation outfeed.
OutfeedEnqueueTuple Enqueue multiple Tensor values on the computation outfeed.
PartitionedInput <T extends TType > An op that groups a list of partitioned inputs together.
PartitionedInput.Options Optional attributes for PartitionedInput
PartitionedOutput <T extends TType > An op that demultiplexes a tensor to be sharded by XLA to a list of partitioned

outputs outside the XLA computation.

PartitionedOutput.Options Optional attributes for PartitionedOutput
Prelinearize An op which linearizes one Tensor value to an opaque variant tensor.
Prelinearize.Options Optional attributes for Prelinearize
PrelinearizeTuple An op which linearizes multiple Tensor values to an opaque variant tensor.
PrelinearizeTuple.Options Optional attributes for PrelinearizeTuple
RecvTPUEmbeddingActivations An op that receives embedding activations on the TPU.
ReplicatedInput <T extends TType > Connects N inputs to an N-way replicated TPU computation.
ReplicatedInput.Options Optional attributes for ReplicatedInput
ReplicatedOutput <T extends TType > Connects N outputs from an N-way replicated TPU computation.
ReplicateMetadata Metadata indicating how the TPU computation should be replicated.
ReplicateMetadata.Options Optional attributes for ReplicateMetadata
RetrieveTPUEmbeddingAdadeltaParameters Retrieve Adadelta embedding parameters.
RetrieveTPUEmbeddingAdadeltaParameters.Options Optional attributes for RetrieveTPUEmbeddingAdadeltaParameters
RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug Retrieve Adadelta embedding parameters with debug support.
RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug.Options Optional attributes for RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug
RetrieveTPUEmbeddingAdagradParameters Retrieve Adagrad embedding parameters.
RetrieveTPUEmbeddingAdagradParameters.Options Optional attributes for RetrieveTPUEmbeddingAdagradParameters
RetrieveTPUEmbeddingAdagradParametersGradAccumDebug Retrieve Adagrad embedding parameters with debug support.
RetrieveTPUEmbeddingAdagradParametersGradAccumDebug.Options Optional attributes for RetrieveTPUEmbeddingAdagradParametersGradAccumDebug
RetrieveTPUEmbeddingADAMParameters Retrieve ADAM embedding parameters.
RetrieveTPUEmbeddingADAMParameters.Options Optional attributes for RetrieveTPUEmbeddingADAMParameters
RetrieveTPUEmbeddingADAMParametersGradAccumDebug Retrieve ADAM embedding parameters with debug support.
RetrieveTPUEmbeddingADAMParametersGradAccumDebug.Options Optional attributes for RetrieveTPUEmbeddingADAMParametersGradAccumDebug
RetrieveTPUEmbeddingCenteredRMSPropParameters Retrieve centered RMSProp embedding parameters.
RetrieveTPUEmbeddingCenteredRMSPropParameters.Options Optional attributes for RetrieveTPUEmbeddingCenteredRMSPropParameters
RetrieveTPUEmbeddingFTRLParameters Retrieve FTRL embedding parameters.
RetrieveTPUEmbeddingFTRLParameters.Options Optional attributes for RetrieveTPUEmbeddingFTRLParameters
RetrieveTPUEmbeddingFTRLParametersGradAccumDebug Retrieve FTRL embedding parameters with debug support.
RetrieveTPUEmbeddingFTRLParametersGradAccumDebug.Options Optional attributes for RetrieveTPUEmbeddingFTRLParametersGradAccumDebug
RetrieveTPUEmbeddingMDLAdagradLightParameters Retrieve MDL Adagrad Light embedding parameters.
RetrieveTPUEmbeddingMDLAdagradLightParameters.Options Optional attributes for RetrieveTPUEmbeddingMDLAdagradLightParameters
RetrieveTPUEmbeddingMomentumParameters Retrieve Momentum embedding parameters.
RetrieveTPUEmbeddingMomentumParameters.Options Optional attributes for RetrieveTPUEmbeddingMomentumParameters
RetrieveTPUEmbeddingMomentumParametersGradAccumDebug Retrieve Momentum embedding parameters with debug support.
RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.Options Optional attributes for RetrieveTPUEmbeddingMomentumParametersGradAccumDebug
RetrieveTPUEmbeddingProximalAdagradParameters Retrieve proximal Adagrad embedding parameters.
RetrieveTPUEmbeddingProximalAdagradParameters.Options Optional attributes for RetrieveTPUEmbeddingProximalAdagradParameters
RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug Retrieve proximal Adagrad embedding parameters with debug support.
RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug.Options Optional attributes for RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug
RetrieveTPUEmbeddingProximalYogiParameters
RetrieveTPUEmbeddingProximalYogiParameters.Options Optional attributes for RetrieveTPUEmbeddingProximalYogiParameters
RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug
RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug.Options Optional attributes for RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug
RetrieveTPUEmbeddingRMSPropParameters Retrieve RMSProp embedding parameters.
RetrieveTPUEmbeddingRMSPropParameters.Options Optional attributes for RetrieveTPUEmbeddingRMSPropParameters
RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug Retrieve RMSProp embedding parameters with debug support.
RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug.Options Optional attributes for RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug
RetrieveTPUEmbeddingStochasticGradientDescentParameters Retrieve SGD embedding parameters.
RetrieveTPUEmbeddingStochasticGradientDescentParameters.Options Optional attributes for RetrieveTPUEmbeddingStochasticGradientDescentParameters