Lines Matching defs:device

31 def _generate_input(shape, dtype, device, with_extremal):  argument
101 def _test_dim_keepdim(self, op: ReductionOpInfo, device, *, ndim, **dim_keepdim): argument
117 def test_dim_default(self, device, op: ReductionOpInfo): argument
123 def test_dim_default_keepdim(self, device, op: ReductionOpInfo): argument
129 def test_dim_none(self, device, op: ReductionOpInfo): argument
135 def test_dim_none_keepdim(self, device, op: ReductionOpInfo): argument
141 def test_dim_single(self, device, op: ReductionOpInfo): argument
149 def test_dim_single_keepdim(self, device, op: ReductionOpInfo): argument
157 def test_dim_empty(self, device, op: ReductionOpInfo): argument
163 def test_dim_empty_keepdim(self, device, op: ReductionOpInfo): argument
169 def test_dim_multi(self, device, op: ReductionOpInfo): argument
175 def test_dim_multi_keepdim(self, device, op: ReductionOpInfo): argument
181 def test_dim_multi_unsorted(self, device, op: ReductionOpInfo): argument
186 def test_dim_multi_unsorted_keepdim(self, device, op: ReductionOpInfo): argument
191 def test_dim_multi_duplicate(self, device, op: ReductionOpInfo): argument
197 def test_dim_multi_unsupported(self, device, op: ReductionOpInfo): argument
203 def test_dim_offbounds(self, device, op: ReductionOpInfo): argument
209 def test_dim_ndim_limit(self, device, op: ReductionOpInfo): argument
217 def test_identity(self, device, dtype, op: ReductionOpInfo): argument
233 def test_nan_policy_propagate(self, device, dtype, op: ReductionOpInfo): argument
243 def test_nan_policy_omit(self, device, dtype, op: ReductionOpInfo): argument
253 def test_result_dtype(self, device, dtype, op: ReductionOpInfo): argument
276 def test_empty_tensor_empty_slice(self, device, op: ReductionOpInfo): argument
309 def test_empty_tensor_nonempty_slice(self, device, op: ReductionOpInfo): argument
330 def test_noncontiguous_innermost(self, device, dtype, op: ReductionOpInfo): argument
336 def test_noncontiguous_outermost(self, device, dtype, op: ReductionOpInfo): argument
342 def test_noncontiguous_all(self, device, dtype, op: ReductionOpInfo): argument
348 def test_noncontiguous_transposed(self, device, dtype, op: ReductionOpInfo): argument
354 def test_noncontiguous_expanded(self, device, dtype, op: ReductionOpInfo): argument
375 def test_ref_scalar_input(self, device, dtype, op: ReductionOpInfo): argument
381 def test_ref_small_input(self, device, dtype, op: ReductionOpInfo): argument
390 def test_ref_large_input_1D(self, device, dtype, op: ReductionOpInfo): argument
396 def test_ref_large_input_2D(self, device, dtype, op: ReductionOpInfo): argument
404 def test_ref_large_input_64bit_indexing(self, device, dtype, op: ReductionOpInfo): argument
410 def test_ref_duplicate_values(self, device, dtype, op: ReductionOpInfo): argument
420 def test_ref_extremal_values(self, device, dtype, op: ReductionOpInfo): argument
432 def test_var_unbiased(self, device): argument
451 def test_var_stability(self, device): argument
456 def test_sum_dim_reduction_uint8_overflow(self, device): argument
466 def test_dim_reduction_less_than_64(self, device): argument
479 def test_dim_reduction_lastdim(self, device, dtype): argument
491 def test_logsumexp(self, device, dtype): argument
511 def test_logsumexp_integral_promotion(self, device): argument
522 def test_logcumsumexp_complex(self, device, dtype): argument
631 def test_sum_parallel(self, device): argument
671 def test_max_elementwise(self, device): argument
675 def test_min_elementwise(self, device): argument
678 def test_all_any(self, device): argument
707 def test_all_any_with_dim(self, device): argument
724 def test_numpy_named_args(self, device): argument
839 def test_sum_dim(self, device): argument
846 def test_mean_dim(self, device): argument
854 def test_std_dim(self, device): argument
862 def test_var_dim(self, device): argument
871 def test_logsumexp_dim(self, device): argument
879 def test_mean_int_with_optdtype(self, device): argument
935 def test_sum_integer_upcast(self, device): argument
940 def test_prod_integer_upcast(self, device): argument
945 def test_cumsum_integer_upcast(self, device): argument
949 def test_cumprod_integer_upcast(self, device): argument
953 def test_mode(self, device, dtype): argument
986 def _test_mode_intervals(self, shape, intervals, device, dtype, v=1): argument
1004 def test_mode_large(self, device, dtype): argument
1022 def test_mode_boolean(self, device): argument
1050 def test_mode_wrong_dtype(self, device): argument
1074 def test_mode_wrong_device(self, device): argument
1090 def test_accreal_type(self, device) -> None: argument
1100 def test_var_mean_some_dims(self, device): argument
1117 def test_all_any_empty(self, device): argument
1126 def test_all_issue117215(self, device): argument
1135 def test_max_with_inf(self, device, dtype): argument
1144 def test_min_with_inf(self, device, dtype): argument
1151 def _test_minmax_helper(self, torchfn, reffn, device, dtype, skip_indices=False): argument
1152 def create_input(shape, device, dtype): argument
1203 def test_max(self, device, dtype): argument
1209 def test_min(self, device, dtype): argument
1215 def test_amin(self, device, dtype): argument
1221 def test_amax(self, device, dtype): argument
1227 def test_aminmax(self, device, dtype): argument
1240 def test_invalid_0dim_aminmax(self, device, dtype): argument
1246 def test_bincount(self, device): argument
1339 def test_var_stability2(self, device): argument
1354 def test_sum_noncontig_lowp(self, device, dtype) -> None: argument
1372 def helper(self, shape, reduce_dims, device, dtype): argument
1398 def test_sum_all(self, device, dtype) -> None: argument
1410 def _test_memory_format_transformations(self, device, input_generator_fn, transformation_fn, argument
1459 def test_sum_out(self, device, dtype: torch.dtype) -> None: argument
1473 def test_prod_gpu(self, device, dtype): argument
1488 def test_prod(self, device, dtype): argument
1497 def test_prod_lowp(self, device, dtype): argument
1507 def test_prod_bool(self, device): argument
1525 def test_max_mixed_devices(self, device): argument
1536 def test_min_mixed_devices(self, device): argument
1547 def test_bucketization(self, device): argument
1671 def test_nansum(self, device, dtype): argument
1696 def _test_reduction_function_with_numpy(self, torch_func, np_func, device, dtype, argument
1724 def test_count_nonzero(self, device, dtype): argument
1742 …def _test_sum_reduction_vs_numpy(self, torch_fn, np_fn, device, dtype, with_keepdim=False, with_ex… argument
1767 def test_sum_vs_numpy(self, device, dtype): argument
1774 def test_nansum_vs_numpy(self, device, dtype): argument
1781 def test_nansum_complex(self, device, dtype): argument
1787 def test_nansum_out_dtype(self, device, dtype): argument
1801 def test_argminmax_multiple(self, device, dtype): argument
1886 def test_all_any_vs_numpy(self, device, dtype): argument
1998 def test_repeated_dim(self, device): argument
2011 def test_var(self, device): argument
2027 def test_var_large_input(self, device): argument
2037 def test_sum_noncontig(self, device, dtype): argument
2046 def test_min_max_nan(self, device): argument
2066 def test_sum_cpu_device_mismatch(self, device): argument
2083 def test_minmax_illegal_dtype(self, device): argument
2108 def test_dim_arg_reduction_scalar(self, device, dtype): argument
2126 def test_dim_reduction(self, device, dtype): argument
2248 def test_dim_reduction_fns(self, device, dtype, fn_name): argument
2306 def test_reduction_split(self, device): argument
2316 def test_reduction_vectorize_along_input_corner(self, device, dtype): argument
2414 def test_reduction_vectorize_along_output(self, device, dtype): argument
2437 def test_argminmax_large_axis(self, device): argument
2447 def test_argminmax_axis_with_dim_one(self, device): argument
2465 def test_median_real_values(self, device, dtype): argument
2494 def test_median_nan_values(self, device, dtype): argument
2532 def test_median_corner_cases(self, device): argument
2583 def test_quantile(self, device, dtype): argument
2624 def test_quantile_backward(self, device): argument
2640 def test_quantile_error(self, device): argument
2669 def test_std_mean(self, device): argument
2680 def test_std_mean_all_dims(self, device): argument
2689 def test_var_mean(self, device): argument
2700 def test_var_mean_all_dims(self, device): argument
2709 def test_std_mean_some_dims(self, device): argument
2724 def _compare_std_var_with_numpy(self, op, device, dtype, input, dim, argument
2763 def test_var_vs_numpy(self, device, dtype): argument
2774 def test_std_vs_numpy(self, device, dtype): argument
2785 def test_var_correction_vs_numpy(self, device, dtype): argument
2819 def test_std_correction_vs_numpy(self, device, dtype): argument
2853 def test_std_mean_correction(self, device, dtype): argument
2884 def test_var_mean_correction(self, device, dtype): argument
2915 def test_warn_invalid_degrees_of_freedom(self, device, dtype): argument
2927 def test_amin_amax_some_dims(self, device): argument
2947 def test_histc(self, device): argument
3072 def test_histc_lowp(self, device, dtype): argument
3138 def test_histogram(self, device, dtype): argument
3258 def test_histogramdd(self, device, dtype): argument
3316 def test_histogram_error_handling(self, device, dtype): argument
3379 def test_tensor_compare_ops_empty(self, device): argument
3421 def test_tensor_compare_ops_argmax_argmix_kthvalue_dim_empty(self, device): argument
3461 def test_tensor_reduce_ops_empty(self, device): argument
3525 def test_reduction_empty_any_all(self, device): argument
3553 def test_reduce_dtype(self, device): argument
3579 def test_reference_masked(self, device, dtype, op): argument
3618 def test_reductions_large_half_tensors(self, device, dtype): argument