| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_conditional_accumulator_test.py | 444 grad_values=np.array([1, 2]).astype(np.float32)).run()
455 grad_indices=[0], grad_values=np.array(1).astype(np.float32)).run()
467 grad_values=np.array([[0, 1, 1]]).astype(np.float32)).run()
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/core/api_def/base_api/ |
| D | api_def_SparseFillEmptyRowsGrad.pbtxt | 10 name: "grad_values"
29 Takes vectors reverse_index_map, shaped `[N]`, and grad_values,
34 d_values[j] = grad_values[reverse_index_map[j]]
36 grad_values[k] * 1{k not in reverse_index_map})
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | optimizers.py | 276 grad_values = gradient.values
278 grad_values = gradient
280 if grad_values is not None:
283 summary.histogram("gradients/%s" % var_name, grad_values)
286 clip_ops.global_norm([grad_values]))
438 grad_values = grad.values * multiplier
439 grad = ops.IndexedSlices(grad_values, grad.indices, grad.dense_shape)
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | training.py | 283 grad_values = grad.values
285 grad_values = grad
287 summary.histogram(var.op.name + '_gradient', grad_values))
290 clip_ops.global_norm([grad_values])))
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | shampoo.py | 162 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,
176 return self._apply_gradient(grad_values, var, grad_indices)
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/ |
| D | learning.py | 357 grad_values = grad.values
359 grad_values = grad
361 summary.histogram(var.op.name + '/gradient', grad_values))
364 clip_ops.global_norm([grad_values])))
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_fill_empty_rows_op.cc | 218 const auto grad_values = grad_values_t->vec<T>(); in Compute() local
244 d_values(i) = grad_values(reverse_index_map(i)); in Compute()
252 d_default_value += grad_values(j); in Compute()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.sparse.-sparse-conditional-accumulator.pbtxt | 24 …argspec: "args=[\'self\', \'grad_indices\', \'grad_values\', \'grad_shape\', \'local_step\', \'nam…
|
| D | tensorflow.-sparse-conditional-accumulator.pbtxt | 24 …argspec: "args=[\'self\', \'grad_indices\', \'grad_values\', \'grad_shape\', \'local_step\', \'nam…
|
| D | tensorflow.raw_ops.pbtxt | 3493 …argspec: "args=[\'reverse_index_map\', \'grad_values\', \'name\'], varargs=None, keywords=None, de…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.sparse.-sparse-conditional-accumulator.pbtxt | 24 …argspec: "args=[\'self\', \'grad_indices\', \'grad_values\', \'grad_shape\', \'local_step\', \'nam…
|
| D | tensorflow.raw_ops.pbtxt | 3493 …argspec: "args=[\'reverse_index_map\', \'grad_values\', \'name\'], varargs=None, keywords=None, de…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_ops.cc | 606 ShapeHandle grad_values = c->input(1); in __anona7ffb82a1702() local
607 TF_RETURN_IF_ERROR(c->WithRank(grad_values, 1, &grad_values)); in __anona7ffb82a1702()
|
| D | ops.pbtxt | 36556 name: "grad_values"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_grad.py | 309 reverse_index_map=reverse_index_map, grad_values=output_grad_values)
|
| D | data_flow_ops.py | 1384 grad_values=grad.values,
1391 grad_values, argument
1431 gradient_values=grad_values,
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 26954 func SparseFillEmptyRowsGrad(scope *Scope, reverse_index_map tf.Output, grad_values tf.Output) (d_v…
26961 reverse_index_map, grad_values,
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 76152 name: "grad_values"
|
| D | ops_history.v1.pbtxt | 76779 name: "grad_values"
|