Lines Matching defs:device
86 def test_float_domains(self, device, dtype, op): argument
273 def test_reference_numerics_normal(self, device, dtype, op): argument
281 def test_reference_numerics_small(self, device, dtype, op): argument
292 def test_reference_numerics_large(self, device, dtype, op): argument
306 def test_reference_numerics_extremal(self, device, dtype, op): argument
314 def test_contig_vs_every_other(self, device, dtype, op): argument
330 def test_contig_vs_transposed(self, device, dtype, op): argument
346 def test_non_contig(self, device, dtype, op): argument
362 def test_non_contig_index(self, device, dtype, op): argument
380 def test_non_contig_expand(self, device, dtype, op): argument
401 def test_contig_size1(self, device, dtype, op): argument
416 def test_contig_size1_large_dim(self, device, dtype, op): argument
437 def test_batch_vs_slicing(self, device, dtype, op): argument
454 def test_nan_to_num(self, device, dtype): argument
494 def test_nan_to_num_bfloat16(self, device): argument
519 def test_nan_to_num_complex(self, device, dtype): argument
537 def test_complex_edge_values(self, device, dtype): argument
556 def test_digamma_special(self, device, dtype): argument
588 def test_digamma(self, device, dtype): argument
611 def test_frexp(self, device, dtype): argument
623 def test_frexp_assert_raises(self, device): argument
659 def test_polygamma_neg(self, device): argument
667 def test_op_invert(self, device): argument
682 def test_abs_angle_complex_to_float(self, device, dtype): argument
756 self, inplace_op, num_inputs, dtype, device, expected_failure=False argument
795 def test_unary_out_op_mem_overlap(self, device, dtype): argument
906 def test_hardshrink(self, device, dtype): argument
928 def test_hardshrink_edge_cases(self, device, dtype) -> None: argument
961 def test_exp_slow(self, device, dtype): argument
973 def test_hardswish(self, device, dtype): argument
996 def test_hardsigmoid(self, device, dtype): argument
1019 def test_hardsigmoid_backward(self, device, dtype): argument
1032 def test_silu(self, device, dtype): argument
1073 def test_silu_complex(self, device, dtype): argument
1099 def test_sinc(self, device, dtype): argument
1115 def test_mish(self, device, dtype): argument
1156 def test_log1p_complex(self, device, dtype): argument
1213 def test_threshold(self, device, dtype): argument
1224 def _helper_test_igamma(self, loglo, loghi, device, dtype, torch_fcn, scipy_fcn): argument
1253 def test_igamma_common(self, device, dtype): argument
1265 def test_igammac_common(self, device, dtype): argument
1276 def test_igamma_edge_cases(self, device, dtype): argument
1305 def test_igammac_edge_cases(self, device, dtype): argument
1343 def _i0_range_helper(self, range, device, dtype): argument
1354 def test_i0_range1(self, device, dtype): argument
1362 def test_i0_range2(self, device, dtype): argument
1369 def test_i0_range3(self, device, dtype): argument
1377 def test_i0_special(self, device, dtype): argument
1387 def test_special_i0_i1_vs_scipy(self, device, dtype): argument
1430 def test_special_ndtr_vs_scipy(self, device, dtype): argument
1449 def test_special_log_ndtr_vs_scipy(self, device, dtype): argument
1464 def test_abs_big_number(self, device, dtype): argument
1471 def test_abs_signed_zero(self, device, dtype): argument
1484 def test_abs_zero(self, device, dtype): argument
1491 def test_isposinf_isneginf_non_boolean_output(self, device, dtype): argument
1503 def test_nonzero_empty(self, device): argument
1536 def test_exp(self, device, dtype): argument