Searched defs:ReductionOpInfo (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_reductions.py | 101 def _test_dim_keepdim(self, op: ReductionOpInfo, device, *, ndim, **dim_keepdim):
117 def test_dim_default(self, device, op: ReductionOpInfo):
123 def test_dim_default_keepdim(self, device, op: ReductionOpInfo):
129 def test_dim_none(self, device, op: ReductionOpInfo):
135 def test_dim_none_keepdim(self, device, op: ReductionOpInfo):
141 def test_dim_single(self, device, op: ReductionOpInfo):
149 def test_dim_single_keepdim(self, device, op: ReductionOpInfo):
157 def test_dim_empty(self, device, op: ReductionOpInfo):
163 def test_dim_empty_keepdim(self, device, op: ReductionOpInfo):
169 def test_dim_multi(self, device, op: ReductionOpInfo):
[all …]
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/ |
| H A D | core.py | 1536 class ReductionOpInfo(OpInfo): class
|