| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | scatter_ops_test.py | 108 state_ops.scatter_update: _NumpyUpdate,
118 state_ops.scatter_update: _NumpyUpdateScalar,
204 self._VariableRankTests(state_ops.scatter_update, False)
243 self._VariableRankTests(state_ops.scatter_update, False, True)
285 update0 = state_ops.scatter_update(var, 1, True)
286 update1 = state_ops.scatter_update(
|
| D | resource_variable_ops_test.py | 677 v.scatter_update(
733 state_ops.scatter_update(ref, indices, updates)
1264 state_ops.scatter_update(v, [1], [3.0])
1285 v.scatter_update(
1334 state_ops.scatter_update(v, [1], [3])
1344 state_ops.scatter_update(v, [0, 1], [0, 1, 2])
|
| D | control_flow_ops_py_test.py | 2662 ns = state_ops.scatter_update(select, j, 10.0)
2687 ns1 = state_ops.scatter_update(select1, j, 10.0)
2688 ns2 = state_ops.scatter_update(select2, j, 10.0)
2714 ns = state_ops.scatter_update(select, j, 10.0)
|
| D | list_ops_test.py | 964 updated_v = state_ops.scatter_update(v, [3, 5], [l_new_0, l_new_1])
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_values.py | 248 def scatter_update(self, sparse_delta, use_locking=False, name=None): member in TPUDistributedVariable
250 return self._primary.scatter_update(sparse_delta, use_locking, name)
251 return self._policy.scatter_update(
326 def scatter_update(self, *args, **kwargs): member in TPUMirroredVariable
328 return self._primary.scatter_update(*args, **kwargs)
471 def scatter_update(self, *args, **kwargs): member in TPUOnWritePolicy
532 def scatter_update(self, *args, **kwargs): member in TPUOnReadPolicy
|
| D | values.py | 823 def scatter_update(self, sparse_delta, use_locking=False, name=None): member in DistributedVariable
825 return self._primary.scatter_update(sparse_delta, use_locking, name)
827 return self._policy.scatter_update(
829 return values_util.scatter_update(
1057 def scatter_update(self, *args, **kwargs): member in MirroredVariable
1059 return self._primary.scatter_update(*args, **kwargs)
1064 return super(MirroredVariable, self).scatter_update(*args, **kwargs)
1227 def scatter_update(self, *args, **kwargs): member in SyncOnReadVariable
1229 return self._primary.scatter_update(*args, **kwargs)
1489 def scatter_update(self, *args, **kwargs): member in OnReadPolicy
[all …]
|
| D | packed_distributed_variable.py | 226 def scatter_update(self, sparse_delta, use_locking=False, name=None): member in PackedDistributedVariable
227 scatter_update_fn = lambda var, *a, **kw: var.scatter_update(*a, **kw)
330 def scatter_update(self, sparse_delta, use_locking=False, name=None): member in PackedVarAndDevice
332 return self._var.scatter_update(sparse_delta, use_locking, name)
|
| D | values_util.py | 279 def scatter_update(var, sparse_delta, use_locking=False, name=None): function
280 scatter_update_fn = lambda var, *a, **kw: var.scatter_update(*a, **kw)
|
| D | vars_test.py | 608 def scatter_update(v): function
613 return v.scatter_update(value)
618 distribution.run(scatter_update, args=(v1,))))
622 distribution.run(scatter_update, args=(v2,))))
|
| D | values_test.py | 754 w.scatter_update(
867 _assert_unsaveable(lambda: v.scatter_update(_make_index_slices([1.], [0])))
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | state_ops.py | 256 def scatter_update(ref, indices, updates, use_locking=True, name=None): function
302 return gen_state_ops.scatter_update(ref, indices, updates,
|
| D | standard_ops.py | 97 from tensorflow.python.ops.state_ops import scatter_update
|
| D | variables.py | 728 def scatter_update(self, sparse_delta, use_locking=False, name=None): member in Variable
2252 def scatter_update(self, sparse_delta, use_locking=False, name=None): member in RefVariable
2269 return gen_state_ops.scatter_update(
|
| D | resource_variable_ops.py | 1052 def scatter_update(self, sparse_delta, use_locking=False, name=None): member in BaseResourceVariable
2101 def scatter_update(self, sparse_delta, use_locking=False, name=None): member in _UnreadVariable
2104 self).scatter_update(sparse_delta, use_locking, name)
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ScatterNdUpdate.pbtxt | 74 See also `tf.scatter_update` and `tf.batch_scatter_update`.
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | autocast_variable.py | 288 def scatter_update(self, sparse_delta, use_locking=False, name=None): member in AutoCastVariable
289 return self._apply_update(self._variable.scatter_update, sparse_delta,
|
| D | autocast_variable_test.py | 215 evaluate(x.scatter_update(slices(4., 1))), [7, 4])
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-variable.pbtxt | 156 name: "scatter_update"
|
| D | tensorflow.pbtxt | 2040 name: "scatter_update"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-variable.pbtxt | 155 name: "scatter_update"
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | scatter_nd_ops_test.py | 328 scatter_update = state_ops.scatter_nd_update(ref, indices, updates)
329 self.assertAllEqual(scatter_update.get_shape().as_list(), shape)
334 self.assertAllEqual(expected_result, self.evaluate(scatter_update))
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tensor_tracer.py | 736 return state_ops.scatter_update(cache, indices, updates).op
|
| /external/tensorflow/ |
| D | RELEASE.md | 4213 * Improve speed of scatter_update on the cpu for ASSIGN operations.
4663 * `tf.scatter_update` now no longer guarantees that lexicographically largest
|