Class RetrieveAllTPUEmbeddingParameters

java.lang.Object
org.tensorflow.op.RawOp
org.tensorflow.op.tpu.RetrieveAllTPUEmbeddingParameters
All Implemented Interfaces:
Op
@Operator(group="tpu") public final class RetrieveAllTPUEmbeddingParameters extends RawOp
An op that retrieves optimization parameters from embedding to host memory. An op that retrieves optimization parameters from embedding to host memory. Must be preceded by a ConfigureTPUEmbeddingHost op that sets up the correct embedding table configuration. For example, this op is used to retrieve updated parameters before saving a checkpoint. For Adagrad, auxiliary1 will contain the accumulators after running this op. For SGD, all of the auxiliary* values will be empty (0x0 tensors for that table). For FTRL, auxiliary1 will contain the accumulators and auxiliary2 will contain the linear terms. For ADAM, auxiliary1 will contain the momenta and auxiliary2 will contain the velocities.

  • Field Details

  • Constructor Details

    • RetrieveAllTPUEmbeddingParameters

      public RetrieveAllTPUEmbeddingParameters(Operation operation)

  • Method Details

    • create

      @Endpoint(describeByClass=true) public static RetrieveAllTPUEmbeddingParameters create(Scope scope, Long NumTables, String config, Long numShards, Long shardId)
      Factory method to create a class wrapping a new RetrieveAllTPUEmbeddingParameters operation.
      Parameters:
      scope - current scope
      NumTables - The number of embedding tables.
      config - An TPUEmbeddingConfiguration proto describing the table parameters being loaded, serialized to a string.
      numShards - Number of shards into which the embedding tables are divided.
      shardId - Identifier of shard for this operation.
      Returns:
      a new instance of RetrieveAllTPUEmbeddingParameters

    • parameters

      public List<Output<TFloat32>> parameters()
      Gets parameters. A list of tensors, one for each embedding table, containing the stored embedding table parameters.
      Returns:
      parameters.

    • auxiliary1

      public List<Output<TFloat32>> auxiliary1()
      Gets auxiliary1. A list of tensors, one for each embedding table, containing the first auxiliary optimization parameter stored. Elements are present in the list, but have zero size, for unused optimization parameters (based on the algorithm in use for each table).
      Returns:
      auxiliary1.

    • auxiliary2

      public List<Output<TFloat32>> auxiliary2()
      Gets auxiliary2. A list of tensors, one for each embedding table, containing the second auxiliary optimization parameter stored. Elements are present in the list, but have zero size, for unused optimization parameters (based on the algorithm in use for each table).
      Returns:
      auxiliary2.

    • auxiliary3

      public List<Output<TFloat32>> auxiliary3()
      Gets auxiliary3. A list of tensors, one for each embedding table, containing the third auxiliary optimization parameter stored. Elements are present in the list, but have zero size, for unused optimization parameters (based on the algorithm in use for each table).
      Returns:
      auxiliary3.

    • auxiliary4

      public List<Output<TFloat32>> auxiliary4()
      Gets auxiliary4. A list of tensors, one for each embedding table, containing the fourth auxiliary optimization parameter stored. Elements are present in the list, but have zero size, for unused optimization parameters (based on the algorithm in use for each table).
      Returns:
      auxiliary4.

    • auxiliary5

      public List<Output<TFloat32>> auxiliary5()
      Gets auxiliary5. A list of tensors, one for each embedding table, containing the fifth auxiliary optimization parameter stored. Elements are present in the list, but have zero size, for unused optimization parameters (based on the algorithm in use for each table).
      Returns:
      auxiliary5.

    • auxiliary6

      public List<Output<TFloat32>> auxiliary6()
      Gets auxiliary6. A list of tensors, one for each embedding table, containing the six auxiliary optimization parameter stored. Elements are present in the list, but have zero size, for unused optimization parameters (based on the algorithm in use for each table).
      Returns:
      auxiliary6.

    • auxiliary7

      public List<Output<TFloat32>> auxiliary7()
      Gets auxiliary7. A list of tensors, one for each embedding table, containing the seventh auxiliary optimization parameter stored. Elements are present in the list, but have zero size, for unused optimization parameters (based on the algorithm in use for each table).
      Returns:
      auxiliary7.