Searched defs:np_fn (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/test/ |
| D | test_shape_ops.py | 584 def _test_fliplr_flipud(self, torch_fn, np_fn, min_dim, max_dim, device, dtype): argument
|
| 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
|
| D | test_view_ops.py | 1539 def _test_atleast_dim(self, torch_fn, np_fn, device, dtype): argument
|
| D | test_binary_ufuncs.py | 3105 def compare_with_numpy_bin_op(torch_fn, np_fn, x, y, out=None): argument
|
| D | test_linalg.py | 3292 def _test_dot_vdot_vs_numpy(self, device, dtype, torch_fn, np_fn): argument
6767 np_fn = lambda x: np.matmul(x, x) # noqa: E731 function
|
| D | test_reductions.py | 1742 …def _test_sum_reduction_vs_numpy(self, torch_fn, np_fn, device, dtype, with_keepdim=False, with_ex… argument
|
| /external/pytorch/test/xpu/ |
| D | test_gemm.py | 719 np_fn = lambda x: np.matmul(x, x) # noqa: E731 function
|
| /external/pytorch/torch/testing/_internal/ |
| D | common_utils.py | 3751 def compare_with_numpy(self, torch_fn, np_fn, tensor_like, argument
|