Home
last modified time | relevance | path

Searched refs:scatter_reduce_ (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_scatter_gather_ops.py105 if fn is torch.Tensor.scatter_reduce_:
219 … self._test_scatter_base(torch.Tensor.scatter_reduce_, device=device, dtype=dtype,
227 self._test_scatter_base(torch.Tensor.scatter_reduce_, device=device, dtype=dtype,
237 … self._test_scatter_base(torch.Tensor.scatter_reduce_, device=device, dtype=dtype,
245 self._test_scatter_base(torch.Tensor.scatter_reduce_, device=device, dtype=dtype,
253 input.scatter_reduce_(0, idx, src, 'amax', include_self=include_self)
264 self._test_scatter_base(torch.Tensor.scatter_reduce_, device=device, dtype=dtype,
272 input.scatter_reduce_(0, idx, src, 'amin', include_self=include_self)
H A Dtest_sparse_csr.py2641 out.scatter_reduce_(0, index, src, reduce=reduce_type, include_self=False)
/aosp_15_r20/external/pytorch/torch/masked/
H A D_ops.py716 new_values = out.scatter_reduce_(
722 new_values = out.scatter_reduce_(
775 new_values.scatter_reduce_(
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py3399 return scatter_reduce_(self, dim, index, src, reduce)
3409 return scatter_reduce_(x, dim, index, src, "sum")
3414 return scatter_reduce_(clone(x), dim, index, src, reduction_type, **kwargs)
3417 @register_lowering(aten.scatter_reduce_, type_promotion_kind=None)
3418 def scatter_reduce_(self, dim: int, index, src, reduce, *, include_self: bool = True): function
3421 len(aten.scatter_reduce_.overloads()) == 1
3422 and "two" in aten.scatter_reduce_.overloads()
3429 aten.scatter_reduce_.two,
H A Dutils.py1824 in (torch.ops.aten.scatter_reduce_, torch.ops.aten.scatter_reduce)
1841 op_overload.overloadpacket == torch.ops.aten.scatter_reduce_
H A Ddecomposition.py905 aten.scatter_reduce_.two,
/aosp_15_r20/external/pytorch/test/expect/
H A DHasDecompTest.test_aten_core_operators.expect443 aten::scatter_reduce_.two
/aosp_15_r20/external/pytorch/docs/source/
H A Dtensors.rst672 Tensor.scatter_reduce_
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_torchinductor.py7103 a.scatter_reduce_(dim, index, b, reduce=reduce)
7105 a1.scatter_reduce_(dim, index, b, reduce=reduce)
/aosp_15_r20/external/pytorch/torch/_decomp/
H A Ddecompositions.py5112 register_inplace(aten.scatter_reduce_, aten.scatter_reduce)
/aosp_15_r20/external/pytorch/torch/
H A D_meta_registrations.py5694 @register_meta(aten.scatter_reduce_.two)