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 . |