| /external/pytorch/torch/onnx/ |
| D | symbolic_opset16.py | 74 def scatter_add(g: jit_utils.GraphContext, self, dim, index, src): function
|
| D | symbolic_opset9.py | 5086 def scatter_add(g: jit_utils.GraphContext, self, dim, index, src): function
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_values.py | 183 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in TPUDistributedVariable
316 def scatter_add(self, *args, **kwargs): member in TPUMirroredVariable
497 def scatter_add(self, var, sparse_delta, use_locking=False, name=None): member in TPUOnWritePolicy
570 def scatter_add(self, *args, **kwargs): member in TPUOnReadPolicy
|
| D | packed_distributed_variable.py | 182 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in PackedDistributedVariable
322 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in PackedVarAndDevice
|
| D | values_v2.py | 175 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in DistributedVariable
|
| D | values_util.py | 230 def scatter_add(var, sparse_delta, use_locking=False, name=None): function
|
| D | values.py | 816 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in DistributedVariable
1277 def scatter_add(self, *args, **kwargs): member in SyncOnReadVariable
1578 def scatter_add(self, *args, **kwargs): member in OnReadPolicy
1665 def scatter_add(self, var, sparse_delta, use_locking=False, name=None): member in OnWritePolicy
|
| D | sharded_variable.py | 613 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in ShardedVariableMixin
|
| D | vars_test.py | 514 def scatter_add(): function
|
| /external/tensorflow/tensorflow/python/training/ |
| D | adam.py | 247 def _apply_sparse_shared(self, grad, var, indices, scatter_add): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | state_ops.py | 499 def scatter_add(ref, indices, updates, use_locking=False, name=None): function
|
| D | resource_variable_ops.py | 984 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in BaseResourceVariable
2191 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in _UnreadVariable
|
| D | variables.py | 650 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in Variable
2124 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in RefVariable
|
| /external/pytorch/aten/src/ATen/native/ |
| D | NamedTensor.cpp | 374 Tensor scatter_add(const Tensor& self, Dimname dim, const Tensor& index, const Tensor& source) { in scatter_add() function
|
| D | TensorAdvancedIndexing.cpp | 247 TORCH_META_FUNC(scatter_add) in TORCH_META_FUNC() argument
1888 TORCH_IMPL_FUNC(scatter_add) in TORCH_IMPL_FUNC() argument
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | autocast_variable.py | 263 def scatter_add(self, sparse_delta, use_locking=False, name=None): member in AutoCastVariable
|
| /external/pytorch/torch/_inductor/ |
| D | lowering.py | 3403 def scatter_add(x, dim: int, index, src): function
|