Lines Matching defs:device
39 def _generate_input(shape, dtype, device, with_extremal): argument
74 def test_unbind(self, device): argument
88 def test_tolist(self, device): argument
111 def test_movedim_invalid(self, device, dtype): argument
150 def test_movedim(self, device, dtype): argument
232 def test_diag(self, device, dtype): argument
243 def test_diagonal(self, device): argument
256 def test_diagonal_multidim(self, device, dtype): argument
274 def test_trace(self, device, dtype): argument
294 def generate_clamp_baseline(self, device, dtype, *, min_vals, max_vals, with_nans): argument
320 def test_clamp(self, device, dtype): argument
353 def test_clamp_propagates_nans(self, device): argument
391 def test_clamp_raises_arg_errors(self, device): argument
402 def test_flip(self, device, dtype): argument
539 def test_flip_errors(self, device, dtype): argument
556 def test_flip_numpy(self, device, dtype): argument
577 def test_flip_large_tensor(self, device): argument
584 def _test_fliplr_flipud(self, torch_fn, np_fn, min_dim, max_dim, device, dtype): argument
595 def test_fliplr(self, device, dtype): argument
599 def test_fliplr_invalid(self, device, dtype): argument
607 def test_flipud(self, device, dtype): argument
611 def test_flipud_invalid(self, device, dtype): argument
615 def test_rot90(self, device): argument
652 def test_complex_rot90(self, device, dtype): argument
663 def test_nonzero_no_warning(self, device): argument
672 def test_nonzero(self, device, dtype): argument
682 def gen_nontrivial_input(shape, dtype, device): argument
733 def test_nonzero_astuple_out(self, device): argument
767 def test_nonzero_discontiguous(self, device): argument
793 def test_nonzero_non_diff(self, device): argument
799 def test_sparse_dense_dim(self, device, dtype): argument