| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | scatter_ops_test.py | 109 state_ops.scatter_add: _NumpyAdd,
119 state_ops.scatter_add: _NumpyAddScalar,
207 self._VariableRankTests(state_ops.scatter_add, False)
225 self._VariableRankTests(state_ops.scatter_add, True)
246 self._VariableRankTests(state_ops.scatter_add, False, True)
264 self._VariableRankTests(state_ops.scatter_add, True, True)
|
| D | embedding_ops_test.py | 56 def _TestCase(self, shape, indices, scatter_op=state_ops.scatter_add):
84 if scatter_op == state_ops.scatter_add:
128 state_ops.scatter_add(var, indices, values)
133 state_ops.scatter_add(var, indices, values)
|
| D | resource_variable_ops_test.py | 579 v.scatter_add(
1049 assert_eq(var.scatter_add(slices).scatter_add(slices), [1., 6.])
1052 assert_eq(var.scatter_max(slices2).scatter_add(slices), [3., 4.])
1053 assert_eq(var.scatter_add(slices).scatter_min(slices), [3., 2.])
1270 state_ops.scatter_add(v, [1], [3])
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_values.py | 218 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in TPUDistributedVariable
220 return self._primary.scatter_add(sparse_delta, use_locking, name)
221 return self._policy.scatter_add(
301 def scatter_add(self, *args, **kwargs): member in TPUMirroredVariable
303 return self._primary.scatter_add(*args, **kwargs)
456 def scatter_add(self, *args, **kwargs): member in TPUOnWritePolicy
517 def scatter_add(self, *args, **kwargs): member in TPUOnReadPolicy
|
| D | packed_distributed_variable.py | 186 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in PackedDistributedVariable
187 scatter_add_fn = lambda var, *a, **kw: var.scatter_add(*a, **kw)
310 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in PackedVarAndDevice
312 return self._var.scatter_add(sparse_delta, use_locking, name)
|
| D | values.py | 778 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in DistributedVariable
780 return self._primary.scatter_add(sparse_delta, use_locking, name)
782 return self._policy.scatter_add(
784 return values_util.scatter_add(
1202 def scatter_add(self, *args, **kwargs): member in SyncOnReadVariable
1204 return self._primary.scatter_add(*args, **kwargs)
1469 def scatter_add(self, *args, **kwargs): member in OnReadPolicy
1548 def scatter_add(self, var, sparse_delta, use_locking=False, name=None): member in OnWritePolicy
1549 return values_util.scatter_add(var, sparse_delta, use_locking=use_locking,
|
| D | values_util.py | 234 def scatter_add(var, sparse_delta, use_locking=False, name=None): function
235 scatter_add_fn = lambda var, *a, **kw: var.scatter_add(*a, **kw)
|
| D | vars_test.py | 487 def scatter_add(): function
494 return v.scatter_add(value)
498 distribution.run(scatter_add)))
637 self.evaluate(v1.scatter_add(value))
1182 self.evaluate(distribution.run(v.scatter_add, args=(delta,)))
|
| D | values_test.py | 729 w.scatter_add(_make_index_slices(values=[1., 2.], indices=[0, 2])),
861 _assert_unsaveable(lambda: v.scatter_add(_make_index_slices([1.], [0])))
|
| /external/tensorflow/tensorflow/python/training/ |
| D | adam.py | 186 def _apply_sparse_shared(self, grad, var, indices, scatter_add): argument
200 m_t = scatter_add(m, indices, m_scaled_g_values)
206 v_t = scatter_add(v, indices, v_scaled_g_values)
217 lambda x, i, v: state_ops.scatter_add( # pylint: disable=g-long-lambda
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | state_ops.py | 372 def scatter_add(ref, indices, updates, use_locking=False, name=None): function
416 return gen_state_ops.scatter_add(ref, indices, updates,
|
| D | standard_ops.py | 91 from tensorflow.python.ops.state_ops import scatter_add
|
| D | clustering_ops.py | 494 update_counts = state_ops.scatter_add(total_counts, unique_ids,
496 update_cluster_centers = state_ops.scatter_add(
|
| D | variables.py | 646 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in Variable
2130 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in RefVariable
2147 return gen_state_ops.scatter_add(
|
| D | resource_variable_ops.py | 935 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in BaseResourceVariable
2076 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in _UnreadVariable
2078 return super(_UnreadVariable, self).scatter_add(sparse_delta, use_locking,
|
| D | metrics_impl.py | 1081 update_total_op = state_ops.scatter_add(total, labels, ones)
1082 update_count_op = state_ops.scatter_add(count, labels, is_correct)
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | adam.py | 450 m.scatter_add(ops.IndexedSlices(m_scaled_g_values, indices))
456 v.scatter_add(ops.IndexedSlices(v_scaled_g_values, indices))
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | autocast_variable.py | 268 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in AutoCastVariable
269 return self._apply_update(self._variable.scatter_add, sparse_delta,
|
| D | autocast_variable_test.py | 209 self.assertAllEqual(evaluate(x.scatter_add(slices(1., 0))), [7, 8])
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-variable.pbtxt | 120 name: "scatter_add"
|
| D | tensorflow.pbtxt | 2000 name: "scatter_add"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-variable.pbtxt | 119 name: "scatter_add"
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | scatter_nd_ops_test.py | 414 add_result = state_ops.scatter_add(ref, indices, updates)
|