| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | weight_decay_optimizers.py | 179 def _decay_weights_sparse_op(self, var, indices, scatter_add): argument
182 return scatter_add(var, indices, update, self._use_locking)
197 scatter_add = state_ops.scatter_add
198 decay_op = self._decay_weights_sparse_op(var, grad.indices, scatter_add)
211 scatter_add = self._resource_scatter_add
212 decay_op = self._decay_weights_sparse_op(var, indices, scatter_add)
|
| D | nadam_optimizer.py | 71 def _apply_sparse_shared(self, grad, var, indices, scatter_add): argument
85 m_t = scatter_add(m, indices, m_scaled_g_values)
93 v_t = scatter_add(v, indices, v_scaled_g_values)
96 var_update = scatter_add(var, indices, -lr * m_bar / (v_sqrt + epsilon_t))
|
| D | adam_gs_optimizer.py | 179 def _apply_sparse_shared(self, grad, var, indices, scatter_add): argument
193 m_t = scatter_add(m, indices, m_scaled_g_values)
199 v_t = scatter_add(v, indices, v_scaled_g_values)
210 lambda x, i, v: state_ops.scatter_add( # pylint: disable=g-long-lambda
|
| D | adamax.py | 138 scatter_add, scatter_update): argument
161 var_update = scatter_add(var, indices, var_slice)
167 lambda x, i, v: state_ops.scatter_add( # pylint: disable=g-long-lambda
|
| D | shampoo.py | 398 mat_g_updated = state_ops.scatter_add(mat_g, indices,
423 var_updated = state_ops.scatter_add(
|
| /external/tensorflow/tensorflow/contrib/optimizer_v2/ |
| D | adam.py | 149 def _apply_sparse_shared(self, grad, var, indices, scatter_add, state): argument
163 m_t = scatter_add(m, indices, m_scaled_g_values)
169 v_t = scatter_add(v, indices, v_scaled_g_values)
178 lambda x, i, v: state_ops.scatter_add( # pylint: disable=g-long-lambda
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | scatter_ops_test.py | 109 state_ops.scatter_add: _NumpyAdd,
119 state_ops.scatter_add: _NumpyAddScalar,
210 self._VariableRankTests(state_ops.scatter_add, False)
234 self._VariableRankTests(state_ops.scatter_add, True)
262 self._VariableRankTests(state_ops.scatter_add, False, True)
286 self._VariableRankTests(state_ops.scatter_add, True, True)
|
| D | embedding_ops_test.py | 55 def _TestCase(self, shape, indices, scatter_op=state_ops.scatter_add):
83 if scatter_op == state_ops.scatter_add:
127 state_ops.scatter_add(var, indices, values)
132 state_ops.scatter_add(var, indices, values)
|
| D | resource_variable_ops_test.py | 960 state_ops.scatter_add(v, [1], [3])
|
| /external/tensorflow/tensorflow/python/training/ |
| D | adam.py | 179 def _apply_sparse_shared(self, grad, var, indices, scatter_add): argument
193 m_t = scatter_add(m, indices, m_scaled_g_values)
199 v_t = scatter_add(v, indices, v_scaled_g_values)
210 lambda x, i, v: state_ops.scatter_add( # pylint: disable=g-long-lambda
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | state_ops.py | 368 def scatter_add(ref, indices, updates, use_locking=False, name=None): function
412 return gen_state_ops.scatter_add(ref, indices, updates,
|
| D | standard_ops.py | 86 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 | 613 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in Variable
1860 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in RefVariable
1877 return gen_state_ops.scatter_add(
|
| D | metrics_impl.py | 1053 update_total_op = state_ops.scatter_add(total, labels, ones)
1054 update_count_op = state_ops.scatter_add(count, labels, is_correct)
|
| D | resource_variable_ops.py | 1093 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in ResourceVariable
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-variable.pbtxt | 100 name: "scatter_add"
|
| D | tensorflow.pbtxt | 1916 name: "scatter_add"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-variable.pbtxt | 99 name: "scatter_add"
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | clustering_ops.py | 493 update_counts = state_ops.scatter_add(total_counts, unique_ids,
495 update_cluster_centers = state_ops.scatter_add(
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/ |
| D | sdca_ops.py | 380 result = state_ops.scatter_add(weights[p], gather_ids[p], updates[p])
556 update_ops.append(state_ops.scatter_add(w, i, u))
|
| /external/tensorflow/tensorflow/contrib/metrics/python/ops/ |
| D | metric_ops.py | 1671 update_tp = state_ops.scatter_add(
1673 update_fp = state_ops.scatter_add(
|