Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/kernel_tests/
Dsparse_conditional_accumulator_test.py443 grad_indices=[[0, 1], [1, 0]],
455 grad_indices=[0], grad_values=np.array(1).astype(np.float32)).run()
466 grad_indices=[0, 1],
478 accum_op = q.apply_grad(grad_indices=x_indices, grad_values=x_values)
498 accum_op = q.apply_grad(grad_indices=x_indices, grad_values=x_values)
516 q.apply_grad(grad_indices=0, grad_values=[1.0], grad_shape=[]).run()
520 q.apply_grad(grad_indices=0, grad_values=[1.0]).run()
524 q.apply_grad(grad_indices=[0], grad_values=1.0, grad_shape=[]).run()
528 q.apply_grad(grad_indices=[0], grad_values=1.0).run()
531 q.apply_grad(grad_indices=[0], grad_values=[1.0], grad_shape=[]).run()
[all …]
/external/tensorflow/tensorflow/contrib/opt/python/training/
Dshampoo.py162 def _resource_apply_sparse(self, grad_values, var, grad_indices): argument
163 return self._apply_sparse_shared(grad_values, grad_indices, var)
168 def _apply_sparse_shared(self, grad_values, grad_indices, var): argument
173 array_ops.expand_dims(grad_indices, axis=1), grad_values,
174 array_ops.shape(var, out_type=grad_indices.dtype))
176 return self._apply_gradient(grad_values, var, grad_indices)
Dshampoo_test.py376 grad_indices = np.sort(np.random.choice(np.arange(size[0]), sample_size_1,
392 constant_op.constant(grad_indices),
424 mat_g1_acc[grad_indices] += mat_g1
429 new_val_np[grad_indices, :] -= np.dot(grad_np * mat_left, mat_right)
461 grad_indices = np.sort(np.random.choice(np.arange(size[0]), sample_size,
472 constant_op.constant(grad_indices),
492 grad_dense[grad_indices] = grad_np
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.sparse.-sparse-conditional-accumulator.pbtxt24 …argspec: "args=[\'self\', \'grad_indices\', \'grad_values\', \'grad_shape\', \'local_step\', \'nam…
Dtensorflow.-sparse-conditional-accumulator.pbtxt24 …argspec: "args=[\'self\', \'grad_indices\', \'grad_values\', \'grad_shape\', \'local_step\', \'nam…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.sparse.-sparse-conditional-accumulator.pbtxt24 …argspec: "args=[\'self\', \'grad_indices\', \'grad_values\', \'grad_shape\', \'local_step\', \'nam…
/external/tensorflow/tensorflow/python/ops/
Ddata_flow_ops.py1383 grad_indices=grad.indices,
1390 grad_indices, argument
1430 gradient_indices=math_ops.cast(grad_indices, _dtypes.int64),