| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | scatter_ops_test.py | 107 state_ops.scatter_sub: _NumpySub,
117 state_ops.scatter_sub: _NumpySubScalar,
207 self._VariableRankTests(state_ops.scatter_sub, False)
225 self._VariableRankTests(state_ops.scatter_sub, True)
246 self._VariableRankTests(state_ops.scatter_sub, False, True)
264 self._VariableRankTests(state_ops.scatter_sub, True, True)
|
| D | scatter_nd_ops_test.py | 428 sub_result = state_ops.scatter_sub(ref, indices, updates)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_values.py | 177 def scatter_sub(self, sparse_delta, use_locking=False, name=None): member in TPUDistributedVariable
179 return self._primary.scatter_sub(sparse_delta, use_locking, name)
180 return self._policy.scatter_sub(
311 def scatter_sub(self, *args, **kwargs): member in TPUMirroredVariable
313 return self._primary.scatter_sub(*args, **kwargs)
492 def scatter_sub(self, var, sparse_delta, use_locking=False, name=None): member in TPUOnWritePolicy
567 def scatter_sub(self, *args, **kwargs): member in TPUOnReadPolicy
|
| D | packed_distributed_variable.py | 174 def scatter_sub(self, sparse_delta, use_locking=False, name=None): member in PackedDistributedVariable
175 scatter_sub_fn = lambda var, *a, **kw: var.scatter_sub(*a, **kw)
318 def scatter_sub(self, sparse_delta, use_locking=False, name=None): member in PackedVarAndDevice
320 return self._var.scatter_sub(sparse_delta, use_locking, name)
|
| D | values.py | 807 def scatter_sub(self, sparse_delta, use_locking=False, name=None): member in DistributedVariable
809 return self._primary.scatter_sub(sparse_delta, use_locking, name)
811 return self._policy.scatter_sub(
813 return values_util.scatter_sub(
1272 def scatter_sub(self, *args, **kwargs): member in SyncOnReadVariable
1274 return self._primary.scatter_sub(*args, **kwargs)
1574 def scatter_sub(self, *args, **kwargs): member in OnReadPolicy
1661 def scatter_sub(self, var, sparse_delta, use_locking=False, name=None): member in OnWritePolicy
1662 return values_util.scatter_sub(
|
| D | values_v2.py | 171 def scatter_sub(self, sparse_delta, use_locking=False, name=None): member in DistributedVariable
173 return super().scatter_sub(sparse_delta, use_locking, name)
|
| D | values_util.py | 221 def scatter_sub(var, sparse_delta, use_locking=False, name=None): function
222 scatter_sub_fn = lambda var, *a, **kw: var.scatter_sub(*a, **kw)
|
| D | sharded_variable.py | 663 def scatter_sub(self, sparse_delta, use_locking=False, name=None): member in ShardedVariableMixin
670 v.scatter_sub(per_var_sparse_delta[i], name=new_name)
|
| D | vars_test.py | 489 def scatter_sub(): function
499 return v.scatter_sub(value)
503 distribution.run(scatter_sub)))
1223 self.evaluate(distribution.run(v.scatter_sub, args=(delta,)))
|
| D | distributed_variable_test.py | 424 w.scatter_sub(_make_index_slices(values=[2., 0.5], indices=[0, 1])),
538 _assert_unsaveable(lambda: v.scatter_sub(_make_index_slices([1.], [0])))
|
| D | values_v2_test.py | 171 v.scatter_sub(
|
| /external/tensorflow/tensorflow/python/training/ |
| D | gradient_descent.py | 77 return var.scatter_sub(delta, use_locking=self._use_locking)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | state_ops.py | 614 def scatter_sub(ref, indices, updates, use_locking=False, name=None): function
660 return gen_state_ops.scatter_sub(ref, indices, updates,
|
| D | standard_ops.py | 89 from tensorflow.python.ops.state_ops import scatter_sub
|
| D | variables.py | 634 def scatter_sub(self, sparse_delta, use_locking=False, name=None): member in Variable
2100 def scatter_sub(self, sparse_delta, use_locking=False, name=None): member in RefVariable
2117 return gen_state_ops.scatter_sub(
|
| D | resource_variable_ops.py | 960 def scatter_sub(self, sparse_delta, use_locking=False, name=None): member in BaseResourceVariable
2186 def scatter_sub(self, sparse_delta, use_locking=False, name=None): member in _UnreadVariable
2188 return super(_UnreadVariable, self).scatter_sub(sparse_delta, use_locking,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.__internal__.distribute.-sharded-variable.pbtxt | 73 name: "scatter_sub"
|
| D | tensorflow.-variable.pbtxt | 151 name: "scatter_sub"
|
| D | tensorflow.experimental.dtensor.-d-variable.pbtxt | 188 name: "scatter_sub"
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | autocast_variable.py | 259 def scatter_sub(self, sparse_delta, use_locking=False, name=None): member in AutoCastVariable
260 return self._apply_update(self._variable.scatter_sub, sparse_delta,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-variable.pbtxt | 152 name: "scatter_sub"
|
| D | tensorflow.pbtxt | 2044 name: "scatter_sub"
|
| /external/tensorflow/tensorflow/python/kernel_tests/variables/ |
| D | resource_variable_ops_test.py | 608 v.scatter_sub(
1081 assert_eq(var.scatter_sub(slices).scatter_sub(slices), [1., 2.])
1315 state_ops.scatter_sub(v, [1], [3])
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | embedding_ops_test.py | 113 self._TestCase(_AsLong(list(shape)), list(indices), state_ops.scatter_sub)
|