| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | scatter_ops_test.py | 108 state_ops.scatter_mul: _NumpyMul,
118 state_ops.scatter_mul: _NumpyMulScalar,
210 self._VariableRankTests(state_ops.scatter_mul, False)
228 self._VariableRankTests(state_ops.scatter_mul, True)
249 self._VariableRankTests(state_ops.scatter_mul, False, True)
267 self._VariableRankTests(state_ops.scatter_mul, True, True)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_values.py | 189 def scatter_mul(self, sparse_delta, use_locking=False, name=None): member in TPUDistributedVariable
191 return self._primary.scatter_mul(sparse_delta, use_locking, name)
192 return self._policy.scatter_mul(
331 def scatter_mul(self, *args, **kwargs): member in TPUMirroredVariable
333 return self._primary.scatter_mul(*args, **kwargs)
512 def scatter_mul(self, var, sparse_delta, use_locking=False, name=None): member in TPUOnWritePolicy
579 def scatter_mul(self, *args, **kwargs): member in TPUOnReadPolicy
|
| D | packed_distributed_variable.py | 190 def scatter_mul(self, sparse_delta, use_locking=False, name=None): member in PackedDistributedVariable
191 scatter_mul_fn = lambda var, *a, **kw: var.scatter_mul(*a, **kw)
326 def scatter_mul(self, sparse_delta, use_locking=False, name=None): member in PackedVarAndDevice
328 return self._var.scatter_mul(sparse_delta, use_locking, name)
|
| D | values.py | 825 def scatter_mul(self, sparse_delta, use_locking=False, name=None): member in DistributedVariable
827 return self._primary.scatter_mul(sparse_delta, use_locking, name)
829 return self._policy.scatter_mul(
831 return values_util.scatter_mul(
1282 def scatter_mul(self, *args, **kwargs): member in SyncOnReadVariable
1284 return self._primary.scatter_mul(*args, **kwargs)
1582 def scatter_mul(self, *args, **kwargs): member in OnReadPolicy
1669 def scatter_mul(self, var, sparse_delta, use_locking=False, name=None): member in OnWritePolicy
1670 return values_util.scatter_mul(
|
| D | values_v2.py | 179 def scatter_mul(self, sparse_delta, use_locking=False, name=None): member in DistributedVariable
181 return super().scatter_mul(sparse_delta, use_locking, name)
|
| D | values_util.py | 239 def scatter_mul(var, sparse_delta, use_locking=False, name=None): function
240 scatter_mul_fn = lambda var, *a, **kw: var.scatter_mul(*a, **kw)
|
| D | sharded_variable.py | 653 def scatter_mul(self, sparse_delta, use_locking=False, name=None): member in ShardedVariableMixin
660 v.scatter_mul(per_var_sparse_delta[i], name=new_name)
|
| D | vars_test.py | 556 def scatter_mul(): function
564 return v.scatter_mul(value)
567 test_util.gather(distribution, distribution.run(scatter_mul)))
1277 self.evaluate(distribution.run(v.scatter_mul, args=(delta,)))
|
| D | distributed_variable_test.py | 419 w.scatter_mul(_make_index_slices(values=[2., 0.5], indices=[0, 1])),
539 _assert_unsaveable(lambda: v.scatter_mul(_make_index_slices([1.], [0])))
|
| D | values_v2_test.py | 166 v.scatter_mul(
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | state_ops.py | 732 def scatter_mul(ref, indices, updates, use_locking=False, name=None): function
776 return gen_state_ops.scatter_mul(ref, indices, updates,
|
| D | standard_ops.py | 88 from tensorflow.python.ops.state_ops import scatter_mul
|
| D | variables.py | 700 def scatter_mul(self, sparse_delta, use_locking=False, name=None): member in Variable
2198 def scatter_mul(self, sparse_delta, use_locking=False, name=None): member in RefVariable
2215 return gen_state_ops.scatter_mul(
|
| D | resource_variable_ops.py | 1058 def scatter_mul(self, sparse_delta, use_locking=False, name=None): member in BaseResourceVariable
2206 def scatter_mul(self, sparse_delta, use_locking=False, name=None): member in _UnreadVariable
2208 return super(_UnreadVariable, self).scatter_mul(sparse_delta, use_locking,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.__internal__.distribute.-sharded-variable.pbtxt | 69 name: "scatter_mul"
|
| D | tensorflow.-variable.pbtxt | 135 name: "scatter_mul"
|
| D | tensorflow.experimental.dtensor.-d-variable.pbtxt | 164 name: "scatter_mul"
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | autocast_variable.py | 275 def scatter_mul(self, sparse_delta, use_locking=False, name=None): member in AutoCastVariable
276 return self._apply_update(self._variable.scatter_mul, sparse_delta,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-variable.pbtxt | 136 name: "scatter_mul"
|
| D | tensorflow.pbtxt | 2024 name: "scatter_mul"
|
| /external/tensorflow/tensorflow/python/kernel_tests/variables/ |
| D | resource_variable_ops_test.py | 667 v.scatter_mul(
1085 assert_eq(var.scatter_mul(slices).scatter_mul(slices), [3., 8.])
|