| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | adamax.py | 138 scatter_add, scatter_update): argument
150 m_t = scatter_update(m, indices, m_t_slice)
156 v_t = scatter_update(v, indices, v_t_slice)
169 lambda x, i, v: state_ops.scatter_update( # pylint: disable=g-long-lambda
|
| D | ggt.py | 192 flat_grad_updated = state_ops.scatter_update(
206 flat_grad_updated = state_ops.scatter_update(
230 update_grad_buffer = state_ops.scatter_update(grad_buffer, next_grad_index,
|
| D | lazy_adam_optimizer.py | 62 m_t = state_ops.scatter_update(m, grad.indices,
69 v_t = state_ops.scatter_update(v, grad.indices,
|
| D | lazy_adam_gs_optimizer.py | 62 m_t = state_ops.scatter_update(m, grad.indices,
69 v_t = state_ops.scatter_update(v, grad.indices,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | scatter_ops_test.py | 108 state_ops.scatter_update: _NumpyUpdate,
118 state_ops.scatter_update: _NumpyUpdateScalar,
206 self._VariableRankTests(state_ops.scatter_update, False)
258 self._VariableRankTests(state_ops.scatter_update, False, True)
313 update0 = state_ops.scatter_update(var, 1, True)
314 update1 = state_ops.scatter_update(
|
| D | resource_variable_ops_test.py | 510 state_ops.scatter_update(ref, indices, updates)
954 state_ops.scatter_update(v, [1], [3.0])
975 v.scatter_update(
1012 state_ops.scatter_update(v, [1], [3])
1022 state_ops.scatter_update(v, [0, 1], [0, 1, 2])
|
| D | scatter_nd_ops_test.py | 315 scatter_update = state_ops.scatter_nd_update(ref, indices, updates)
316 self.assertAllEqual(scatter_update.get_shape().as_list(), shape)
321 self.assertAllEqual(expected_result, self.evaluate(scatter_update))
|
| D | control_flow_ops_py_test.py | 2278 ns = state_ops.scatter_update(select, j, 10.0)
2303 ns1 = state_ops.scatter_update(select1, j, 10.0)
2304 ns2 = state_ops.scatter_update(select2, j, 10.0)
2330 ns = state_ops.scatter_update(select, j, 10.0)
|
| D | list_ops_test.py | 930 updated_v = state_ops.scatter_update(v, [3, 5], [l_new_0, l_new_1])
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | factorization_ops.py | 652 def scatter_update(cls, factor, indices, values, sharding_func, name=None): member in WALSModel
658 return state_ops.scatter_update(
672 state_ops.scatter_update(factor[i], sharded_ids[i], sharded_values[
956 update_op = self.scatter_update(
|
| D | wals.py | 265 update_processed_items = state_ops.scatter_update(
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | state_ops.py | 252 def scatter_update(ref, indices, updates, use_locking=True, name=None): function
298 return gen_state_ops.scatter_update(ref, indices, updates,
|
| D | standard_ops.py | 92 from tensorflow.python.ops.state_ops import scatter_update
|
| D | variables.py | 630 def scatter_update(self, sparse_delta, use_locking=False, name=None): member in Variable
1884 def scatter_update(self, sparse_delta, use_locking=False, name=None): member in RefVariable
1901 return gen_state_ops.scatter_update(
|
| D | resource_variable_ops.py | 1114 def scatter_update(self, sparse_delta, use_locking=False, name=None): member in ResourceVariable
|
| /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/tools/api/golden/v1/ |
| D | tensorflow.-variable.pbtxt | 120 name: "scatter_update"
|
| D | tensorflow.pbtxt | 1956 name: "scatter_update"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-variable.pbtxt | 119 name: "scatter_update"
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tensor_tracer.py | 773 return state_ops.scatter_update(cache, indices, updates).op
|
| /external/tensorflow/ |
| D | RELEASE.md | 1540 * Improve speed of scatter_update on the cpu for ASSIGN operations.
1990 * `tf.scatter_update` now no longer guarantees that lexicographically largest
|