Lines Matching defs:ReductionOpInfo

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):
175     def test_dim_multi_keepdim(self, device, op: ReductionOpInfo):
181     def test_dim_multi_unsorted(self, device, op: ReductionOpInfo):
186     def test_dim_multi_unsorted_keepdim(self, device, op: ReductionOpInfo):
191     def test_dim_multi_duplicate(self, device, op: ReductionOpInfo):
197     def test_dim_multi_unsupported(self, device, op: ReductionOpInfo):
203     def test_dim_offbounds(self, device, op: ReductionOpInfo):
209     def test_dim_ndim_limit(self, device, op: ReductionOpInfo):
217     def test_identity(self, device, dtype, op: ReductionOpInfo):
233     def test_nan_policy_propagate(self, device, dtype, op: ReductionOpInfo):
243     def test_nan_policy_omit(self, device, dtype, op: ReductionOpInfo):
253     def test_result_dtype(self, device, dtype, op: ReductionOpInfo):
276     def test_empty_tensor_empty_slice(self, device, op: ReductionOpInfo):
309     def test_empty_tensor_nonempty_slice(self, device, op: ReductionOpInfo):
318     def _test_noncontiguous(self, op: ReductionOpInfo, t: torch.Tensor, **reduction_kwargs):
330     def test_noncontiguous_innermost(self, device, dtype, op: ReductionOpInfo):
336     def test_noncontiguous_outermost(self, device, dtype, op: ReductionOpInfo):
342     def test_noncontiguous_all(self, device, dtype, op: ReductionOpInfo):
348     def test_noncontiguous_transposed(self, device, dtype, op: ReductionOpInfo):
354     def test_noncontiguous_expanded(self, device, dtype, op: ReductionOpInfo):
365     def _test_ref(self, op: ReductionOpInfo, t: torch.Tensor, **reduction_kwargs):
375     def test_ref_scalar_input(self, device, dtype, op: ReductionOpInfo):
381     def test_ref_small_input(self, device, dtype, op: ReductionOpInfo):
390     def test_ref_large_input_1D(self, device, dtype, op: ReductionOpInfo):
396     def test_ref_large_input_2D(self, device, dtype, op: ReductionOpInfo):
404     def test_ref_large_input_64bit_indexing(self, device, dtype, op: ReductionOpInfo):
410     def test_ref_duplicate_values(self, device, dtype, op: ReductionOpInfo):
420     def test_ref_extremal_values(self, device, dtype, op: ReductionOpInfo):