• Home
  • Raw
  • Download

Lines Matching defs:device

194     def test_reference_numerics(self, device, dtype, op):  argument
199 def test_reference_numerics_small_values(self, device, dtype, op): argument
222 def test_reference_numerics_large_values(self, device, dtype, op): argument
239 def test_reference_numerics_extremal_values(self, device, dtype, op): argument
253 def test_broadcasting(self, device, dtype, op): argument
263 def test_scalar_support(self, device, dtype, op): argument
274 def test_contig_vs_every_other(self, device, dtype, op): argument
296 def test_contig_vs_transposed(self, device, dtype, op): argument
318 def test_non_contig(self, device, dtype, op): argument
349 def test_non_contig_index(self, device, dtype, op): argument
375 def test_non_contig_expand(self, device, dtype, op): argument
400 def test_contig_size1(self, device, dtype, op): argument
428 def test_contig_size1_large_dim(self, device, dtype, op): argument
456 def test_batch_vs_slicing(self, device, dtype, op): argument
479 def test_type_promotion(self, device, op): argument
792 def test_not_broadcastable(self, device, dtype, op): argument
817 def test_add_broadcast_empty(self, device): argument
853 def test_addcmul_scalars_as_floats(self, device): argument
866 def test_bitwise_ops(self, device, dtype): argument
918 def test_inplace_division(self, device): argument
926 def test_div_rounding_modes(self, device, dtype): argument
972 def test_floor_div_extremal(self, device, dtype): argument
991 def test_div_rounding_nonfinite(self, device, dtype): argument
1038 def test_divide_by_zero_rounding(self, device, dtype): argument
1062 def test_div_rounding_numpy(self, device, dtype): argument
1108 def test_complex_div_underflow_overflow(self, device, dtype): argument
1144 def test_cross_device_inplace_error_msg(self, device): argument
1154 def test_out_resize_warning(self, device): argument
1192 def test_inplace_dunders(self, device): argument
1205 self, inplace_op, num_inputs, dtype, device, expected_failure=False argument
1240 def binary_check_input_output_mem_overlap(self, op, device, expected_failure=False): argument
1262 def test_binary_op_mem_overlap(self, device, dtype): argument
1369 def test_pow(self, device, dtype): argument
1492 def test_pow_scalar_base(self, device): argument
1502 def test_int_and_float_pow(self, device): argument
1584 def test_pow_inplace_resizing_exception(self, device): argument
1606 def test_int_tensor_pow_neg_ints(self, device): argument
1623 def test_long_tensor_pow_floats(self, device): argument
1631 def test_float_scalar_pow_float_tensor(self, device, dtype): argument
1650 def test_cuda_tensor_pow_scalar_tensor(self, device): argument
1664 def test_cpu_tensor_pow_cuda_scalar_tensor(self, device): argument
1680 def test_pow_cuda_complex_extremal_failing(self, device, dtype): argument
1690 def test_complex_scalar_pow_tensor(self, device, dtype): argument
1721 def test_pow_scalar_type_promotion(self, device): argument
1747 def test_tensor_pow_tensor(self, device): argument
1770 def test_logical_xor_with_nontrivial_alignment(self, device): argument
1788 def test_add_with_tail(self, device, dtype): argument
1854 def test_div_and_floordiv_vs_python(self, device): argument
1891 def test_div_and_floordiv_script_vs_python(self, device): argument
1964 def test_idiv_and_ifloordiv_vs_python(self, device): argument
2079 def test_binary_ops_with_scalars(self, device): argument
2125 def test_maximum_minimum_type_promotion(self, device, dtypes): argument
2140 def test_maximum_minimum_int_and_bool(self, device, dtype): argument
2174 def test_maximum_minimum_float(self, device, dtype): argument
2206 def test_maximum_minimum_float_nan_and_inf(self, device, dtype): argument
2269 def test_maximum_minimum_complex(self, device, dtypes): argument
2291 def test_maximum_minimum_cross_device(self, device): argument
2329 def test_maximum_and_minimum_subgradient(self, device, dtypes): argument
2353 def test_maximum_minimum_forward_ad_float32(self, device): argument
2383 def test_mul_intertype_scalar(self, device, dtype): argument
2399 def test_sub(self, device, dtype): argument
2425 def test_csub(self, device, dtype): argument
2447 def test_min_max_binary_op_nan(self, device, dtype): argument
2489 def test_copysign(self, device, dtypes): argument
2567 def test_copysign_subgradient(self, device, dtypes): argument
2617 def test_div(self, device, dtype): argument
2642 def test_true_divide_out(self, device, dtype): argument
2654 def test_divmul_scalar(self, device, dtype): argument
2673 def test_floor_divide_tensor(self, device, dtype): argument
2687 def test_floor_divide_scalar(self, device, dtype): argument
2700 def test_rdiv(self, device, dtype): argument
2712 def test_fmod_remainder_by_zero_float(self, device, dtype): argument
2723 def test_fmod_remainder_by_zero_integral(self, device, dtype): argument
2751 def test_fmod_remainder(self, device, dtype): argument
2817 def test_remainder_fmod_large_dividend(self, device, dtype): argument
2847 def test_hypot(self, device, dtype): argument
2876 def test_gcd(self, device, dtype): argument
2901 def test_lcm(self, device, dtype): argument
2919 def test_nextafter(self, device, dtype): argument
2943 def test_nextafter_bfloat16(self, device, dtype): argument
2984 def _test_cop(self, torchfn, mathfn, dtype, device): argument
3023 def test_cdiv(self, device, dtype): argument
3028 def test_cremainder(self, device, dtype): argument
3033 def test_cmul(self, device, dtype): argument
3038 def test_cpow(self, device, dtype): argument
3045 def test_floor_divide_zero(self, device, dtype): argument
3053 def test_muldiv_scalar(self, device, dtype): argument
3063 def _generate_input(self, shape, dtype, device, with_extremal): argument
3101 def test_comparison_ops_type_promotion_and_broadcasting(self, device, dtypes): argument
3163 def test_signed_shift(self, device, dtype): argument
3179 def test_shift_limits(self, device, dtype): argument
3231 def test_heaviside(self, device, dtypes): argument
3292 def test_heaviside_cross_device(self, device): argument
3316 def test_heaviside_complex(self, device, dtypes): argument
3343 def _test_logical(self, device, dtypes, op, a_, b_, expected_res_): argument
3364 def test_logical_xor(self, device, dtypes): argument
3375 def test_logical_and(self, device, dtypes): argument
3386 def test_logical_or(self, device, dtypes): argument
3391 def test_remainder_overflow(self, device): argument
3400 def test_rpow(self, device): argument
3409 def test_ldexp(self, device): argument
3446 def test_lerp(self, device, dtype): argument
3474 def test_lerp_lowp(self, device, dtype): argument
3490 def test_lerp_lowp_cpu(self, device, dtype): argument
3505 def _test_logaddexp(self, device, dtype, base2): argument
3559 def test_logaddexp(self, device, dtype): argument
3563 def test_logaddexp2(self, device, dtype): argument
3566 def test_add(self, device): argument
3717 def test_addsub_half_tensor(self, device): argument
3728 def test_sub_typing(self, device): argument
3768 def test_mul(self, device): argument
3795 def test_bool_tensor_comparison_ops(self, device): argument
3834 def test_logical(self, device, dtype): argument
3860 def test_atan2(self, device): argument
3861 def _test_atan2_with_size(size, device): argument
3898 def test_atan2_edgecases(self, device): argument
3899 def _test_atan2(x, y, expected, device, dtype): argument
3917 def test_trapezoid(self, device): argument
3918 def test_dx(sizes, dim, dx, device): argument
3925 def test_x(sizes, dim, x, device): argument
3959 def test_cumulative_trapezoid(self, device): argument
3973 def test_dx(sizes, dim, dx, device): argument
3981 def test_x(sizes, dim, x, device): argument
3992 def test_empty_x(sizes, dim, x, device): argument
4049 def test_pow_scalar_overloads_mem_overlap(self, device, dtype): argument
4068 def test_float_power(self, device, dtypes): argument
4166 def test_float_power_exceptions(self, device): argument
4209 def test_xlogy_xlog1py(self, device, dtypes): argument
4301 def test_xlogy_xlog1py_gradients(self, device, dtype): argument
4316 def test_xlogy_xlog1py_scalar_type_promotion(self, device): argument
4332 def test_xlogy_xlog1py_bfloat16(self, device): argument
4388 def test_zeta(self, device, dtypes): argument
4438 def test_mul_chalf_tensor_and_cpu_scalar(self, device, dtype): argument
4521 def test(self, device, dtype): argument