| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | test_case_gen.py | 104 def make_test_cases_broadcast_two_input_tensor(x, y, cpp_args, torch_args, torch_fn): argument
282 def make_simple_generated_case(*args, torch_fn): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_binary_ufuncs.py | 3105 def compare_with_numpy_bin_op(torch_fn, np_fn, x, y, out=None): argument
4212 def out_variant_helper(torch_fn, x, y): argument
4230 def test_helper(torch_fn, reference_fn, inputs, scalar=None): argument
4282 def test_zeros_special_helper(torch_fn, reference_fn, scalar=False): argument
4333 def _compare_helper(x, y, torch_fn, reference_fn): argument
|
| H A D | test_tensor_creation_ops.py | 800 def _hvd_split_helper(self, torch_fn, np_fn, op_name, inputs, device, dtype, dim): argument
874 def _test_special_stacks(self, dim, at_least_dim, torch_fn, np_fn, device, dtype): argument
2646 def _test_linspace_logspace_complex_helper(self, torch_fn, np_fn, device, dtype): argument
2650 def test_fn(torch_fn, numpy_fn, steps): argument
|
| H A D | test_view_ops.py | 1539 def _test_atleast_dim(self, torch_fn, np_fn, device, dtype): argument
1573 def _test_atleast(self, device, torch_fn): argument
|
| H A D | test_shape_ops.py | 584 def _test_fliplr_flipud(self, torch_fn, np_fn, min_dim, max_dim, device, dtype): argument
|
| H A D | test_linalg.py | 3292 def _test_dot_vdot_vs_numpy(self, device, dtype, torch_fn, np_fn): argument
3338 def _test_dot_vdot_invalid_args(self, device, torch_fn, complex_dtypes=False): argument
6766 torch_fn = lambda x: torch.bmm(x, x) # noqa: E731 function
6770 torch_fn = lambda x: torch.mm(x, x) # noqa: E731 function
|
| H A D | test_unary_ufuncs.py | 1388 def check_equal(t, torch_fn, scipy_fn): argument
|
| H A D | test_reductions.py | 1742 …def _test_sum_reduction_vs_numpy(self, torch_fn, np_fn, device, dtype, with_keepdim=False, with_ex… argument
|
| H A D | test_mps.py | 10304 def _test_atleast(self, device, torch_fn): argument
|
| /aosp_15_r20/external/pytorch/test/xpu/ |
| H A D | test_gemm.py | 718 torch_fn = lambda x: torch.bmm(x, x) # noqa: E731 function
722 torch_fn = lambda x: torch.mm(x, x) # noqa: E731 function
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_unbacked_symints.py | 231 def test_mm_and_friends(self, device, torch_fn, coordinate_descent_tuning): argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_utils.py | 3736 def compare_with_reference(self, torch_fn, ref_fn, sample_input, **kwargs): argument
3751 def compare_with_numpy(self, torch_fn, np_fn, tensor_like, argument
|