TensorFlow Resources JVM API 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.