Class RetrieveAllTPUEmbeddingParameters.Inputs

java.lang.Object
org.tensorflow.op.RawOpInputs<RetrieveAllTPUEmbeddingParameters>
org.tensorflow.op.tpu.RetrieveAllTPUEmbeddingParameters.Inputs
Enclosing class:
RetrieveAllTPUEmbeddingParameters

public static class RetrieveAllTPUEmbeddingParameters.Inputs extends RawOpInputs<RetrieveAllTPUEmbeddingParameters>
  • Field Details

    • config

      public final String config
      An TPUEmbeddingConfiguration proto describing the table parameters being loaded, serialized to a string.
    • numShards

      public final long numShards
      Number of shards into which the embedding tables are divided.
    • shardId

      public final long shardId
      Identifier of shard for this operation.
  • Constructor Details