| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | scatter_nd_ops_test.py | 173 scatter = state_ops.scatter_nd_update(ref, indices, updates)
186 update = state_ops.scatter_nd_update(ref, indices, updates)
198 scatter = state_ops.scatter_nd_update(ref, indices, updates)
211 scatter = state_ops.scatter_nd_update(ref, indices, updates)
225 scatter = state_ops.scatter_nd_update(ref, indices, updates)
234 self._VariableRankTests(_NumpyUpdate, state_ops.scatter_nd_update)
282 state_ops.scatter_nd_update):
310 state_ops.scatter_nd_update(ref, indices,
321 state_ops.scatter_nd_update(res, [[0]], [0.22]).eval()
328 scatter_update = state_ops.scatter_nd_update(ref, indices, updates)
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | state_ops.py | 310 def scatter_nd_update(ref, indices, updates, use_locking=True, name=None): function
364 return gen_state_ops.scatter_nd_update(
914 return scatter_nd_update(
|
| D | standard_ops.py | 104 from tensorflow.python.ops.state_ops import scatter_nd_update
|
| D | variables.py | 884 def scatter_nd_update(self, indices, updates, name=None): member in Variable
2426 def scatter_nd_update(self, indices, updates, name=None): member in RefVariable
2472 return gen_state_ops.scatter_nd_update(
|
| D | resource_variable_ops.py | 1233 def scatter_nd_update(self, indices, updates, name=None): member in BaseResourceVariable
2119 def scatter_nd_update(self, indices, updates, name=None): member in _UnreadVariable
2122 self).scatter_nd_update(indices, updates, name)
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ScatterUpdate.pbtxt | 63 See also `tf.batch_scatter_update` and `tf.scatter_nd_update`.
|
| D | api_def_ResourceScatterNdUpdate.pbtxt | 57 update = tf.scatter_nd_update(ref, indices, updates)
|
| D | api_def_ScatterNdUpdate.pbtxt | 62 update = tf.scatter_nd_update(ref, indices, updates)
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | autocast_variable.py | 304 def scatter_nd_update(self, indices, updates, name=None): member in AutoCastVariable
305 return self._apply_update(self._variable.scatter_nd_update, indices,
|
| D | autocast_variable_test.py | 221 evaluate(x.scatter_nd_update([[0], [1]], [1., 2.])), [1, 2])
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-variable.pbtxt | 148 name: "scatter_nd_update"
|
| D | tensorflow.pbtxt | 2032 name: "scatter_nd_update"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-variable.pbtxt | 147 name: "scatter_nd_update"
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | resource_variable_ops_test.py | 1057 var.scatter_nd_update([[1]], [4.]).scatter_nd_add([[0]], [2.])
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 11696 update = tf.scatter_nd_update(ref, indices, updates)
|