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
RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug	Retrieve SGD embedding parameters with debug support.
RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug.Options	Optional attributes for RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug
SendTPUEmbeddingGradients	Performs gradient updates of embedding tables.
ShutdownDistributedTPU	Shuts down a running distributed TPU system.
TPUCompilationResult	Returns the result of a TPU compilation.
TPUEmbeddingActivations	An op enabling differentiation of TPU Embeddings.
TPUReplicatedInput<T extends TType>	Connects N inputs to an N-way replicated TPU computation.
TPUReplicatedInput.Options	Optional attributes for TPUReplicatedInput
TPUReplicatedOutput<T extends TType>	Connects N outputs from an N-way replicated TPU computation.
TPUReplicateMetadata	Metadata indicating how the TPU computation should be replicated.
TPUReplicateMetadata.Options	Optional attributes for TPUReplicateMetadata
WorkerHeartbeat	Worker heartbeat op.