tf.raw_ops.RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug
tf.raw_ops.RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug(
num_shards, shard_id, table_id=-1, table_name='', config='', name=None
)
| Args |
|---|
num_shards | An int. |
shard_id | An int. |
table_id | An optional int. Defaults to -1. |
table_name | An optional string. Defaults to "". |
config | An optional string. Defaults to "". |
name | A name for the operation (optional). |
| Returns |
|---|
A tuple of Tensor objects (parameters, v, m, gradient_accumulators). |
parameters | A Tensor of type float32. |
v | A Tensor of type float32. |
m | A Tensor of type float32. |
gradient_accumulators | A Tensor of type float32. |