Home
last modified time | relevance | path

Searched refs:grad_indices (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dsparse_conditional_accumulator_test.py458 grad_indices=[[0, 1], [1, 0]],
470 grad_indices=[0], grad_values=np.array(1).astype(np.float32)).run()
481 grad_indices=[0, 1],
493 accum_op = q.apply_grad(grad_indices=x_indices, grad_values=x_values)
513 accum_op = q.apply_grad(grad_indices=x_indices, grad_values=x_values)
531 q.apply_grad(grad_indices=0, grad_values=[1.0], grad_shape=[]).run()
535 q.apply_grad(grad_indices=0, grad_values=[1.0]).run()
539 q.apply_grad(grad_indices=[0], grad_values=1.0, grad_shape=[]).run()
543 q.apply_grad(grad_indices=[0], grad_values=1.0).run()
546 q.apply_grad(grad_indices=[0], grad_values=[1.0], grad_shape=[]).run()
[all …]
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-sparse-conditional-accumulator.pbtxt24 …argspec: "args=[\'self\', \'grad_indices\', \'grad_values\', \'grad_shape\', \'local_step\', \'nam…
Dtensorflow.sparse.-sparse-conditional-accumulator.pbtxt24 …argspec: "args=[\'self\', \'grad_indices\', \'grad_values\', \'grad_shape\', \'local_step\', \'nam…
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_gather_ops.py493 grad_indices = ragged_math_ops.range(
498 values=grad_inner_values, indices=grad_indices,
/external/tensorflow/tensorflow/python/ops/
Ddata_flow_ops.py1465 grad_indices=grad.indices,
1472 grad_indices, argument
1512 gradient_indices=math_ops.cast(grad_indices, _dtypes.int64),