tf.raw_ops.LoadTPUEmbeddingProximalYogiParametersGradAccumDebug(
parameters, v, m, gradient_accumulators, num_shards, shard_id, table_id=-1,
table_name='', config='', name=None
)
Args |
|---|
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.
|
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 |
|---|
|
The created Operation.
|
