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.