Home
last modified time | relevance | path

Searched full:scatter_add_ (Results 1 – 25 of 25) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_scatter_gather_ops.py119 if fn is torch.Tensor.scatter_add_:
197 self._test_scatter_base(torch.Tensor.scatter_add_, device=device, dtype=dtype,
207 res0 = torch.zeros(m, n, device=device, dtype=dtype).scatter_add_(0, idx, src)
208 res1 = torch.zeros(m, n, device=device, dtype=dtype).scatter_add_(1, idx, src)
H A Dtest_torch.py3807 # TODO: remove this after scatter_add_ is deprecated.
3814 input.scatter_add_(0, index, src)
3870 input.scatter_add_(0, index, src)
3889 … res = res.scatter_add_(0, torch.tensor([[0, 1, 2, 0, 0], [2, 0, 0, 1, 2]], device=device), x)
4226 self.assertEqual(shape, y.scatter_add_(dim, ind, y_src).shape)
4231 …self.assertEqual(z, z.scatter_add_(2, torch.empty((2, 3, 0), dtype=torch.int64, device=device), z_…
/aosp_15_r20/external/pytorch/torch/nn/utils/_expanded_weights/
H A Dembedding_expanded_weights.py55 return grad_sample.scatter_add_(
/aosp_15_r20/external/pytorch/functorch/dim/
H A D__init__.py180 # scatter_add_
/aosp_15_r20/external/pytorch/torch/distributions/
H A Dmultinomial.py112 counts.scatter_add_(-1, samples, torch.ones_like(samples))
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dcoverage.yaml531 - aten::scatter_add_
953 aten::scatter_add_: 10
H A Dtensor_ops.py196 x.scatter_add_(0, t, y),
H A Dmodel_ops.yaml318 aten::scatter_add_: 16
/aosp_15_r20/external/pytorch/torch/
H A D_tensor_docs.py4428 :meth:`~torch.Tensor.scatter_add_`.
4502 "scatter_add_",
4504 scatter_add_(dim, index, src) -> Tensor
4542 >>> torch.zeros(3, 5, dtype=src.dtype).scatter_add_(0, index, src)
4547 >>> torch.zeros(3, 5, dtype=src.dtype).scatter_add_(0, index, src)
6521 Out-of-place version of :meth:`torch.Tensor.scatter_add_`
H A D__init__.py1236 * :func:`torch.Tensor.scatter_add_` when called on a CUDA tensor
H A D_meta_registrations.py5009 @register_meta(aten.scatter_add_)
H A D_torch_docs.py9419 Out-of-place version of :meth:`torch.Tensor.scatter_add_`
/aosp_15_r20/external/pytorch/test/expect/
H A DHasDecompTest.test_aten_core_operators.expect442 aten::scatter_add_
/aosp_15_r20/external/pytorch/docs/source/
H A Dtensors.rst670 Tensor.scatter_add_
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorAdvancedIndexing.cpp1009 result.scatter_add_(dim, ep_index, source); in TORCH_IMPL_FUNC()
1659 result.scatter_add_(dim, index, grad); in gather_backward()
1947 count.scatter_add_(dim, index, ones); in TORCH_IMPL_FUNC()
H A DReduceOps.cpp869 return result.scatter_add_(dim, indices, grad); in cummaxmin_backward()
H A Dnative_functions.yaml8186 - func: scatter_add_(Tensor(a!) self, int dim, Tensor index, Tensor src) -> Tensor(a!)
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DScatterGatherKernel.cpp895 "scatter_add_", reduce_add); in scatter_add_cpu_kernel()
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py3404 return scatter_add_(clone(x), dim, index, src)
3407 @register_lowering(aten.scatter_add_, type_promotion_kind=None)
3408 def scatter_add_(x, dim: int, index, src): function
H A Dutils.py749 "aten.scatter_add_",
/aosp_15_r20/external/pytorch/benchmarks/dynamo/microbenchmarks/operator_inp_logs/torchbench_train/
H A Dtimm_efficientdet_training.txt462 Operator: aten.scatter_add_.default
/aosp_15_r20/external/pytorch/torch/_decomp/
H A Ddecompositions.py5111 register_inplace(aten.scatter_add_, aten.scatter_add)
/aosp_15_r20/external/pytorch/test/cpp/lazy/
H A Dtest_lazy_ops.cpp4614 torch::Tensor d = a.scatter_add_(dim, c, b); in TEST_F()
4617 torch::Tensor lazy_d = lazy_a.scatter_add_(dim, lazy_c, lazy_b); in TEST_F()
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_pytorch_onnx_onnxruntime.py4182 return out.scatter_add_(1, index, src)
4207 return out.scatter_add_(1, index, src)
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_torchinductor.py4687 new_node_feat.scatter_add_(