• Home
  • Raw
  • Download

Lines Matching defs:device

39 def _generate_input(shape, dtype, device, with_extremal):  argument
81 def _convert_t(dtype, device): argument
92 def _make_tensor(shape, dtype, device, fill_ones=False) -> torch.Tensor: argument
149 def test_conj_self(self, device, dtype): argument
157 def test_view_dtype_new(self, device, dtype): argument
271 def test_view_dtype_upsize_errors(self, device, dtype): argument
306 def test_view_as_complex(self, device): argument
372 def test_view_as_real(self, device, dtype): argument
398 def test_view_tensor_split(self, device, dtype): argument
409 def test_view_tensor_hsplit(self, device, dtype): argument
419 def test_view_tensor_vsplit(self, device, dtype): argument
429 def test_view_tensor_dsplit(self, device, dtype): argument
439 def test_imag_noncomplex(self, device, dtype): argument
447 def test_real_imag_view(self, device, dtype): argument
478 def test_conj_imag_view(self, device, dtype) -> None: argument
492 def test_conj_view_with_shared_memory(self, device) -> None: argument
509 def test_set_real_imag(self, device, dtypes): argument
526 def test_diagonal_view(self, device) -> None: argument
541 def test_select_view(self, device) -> None: argument
551 def test_unbind_view(self, device) -> None: argument
585 def test_expand_view(self, device) -> None: argument
593 def test_expand_as_view(self, device): argument
602 def test_narrow_view(self, device): argument
610 def test_permute_view(self, device) -> None: argument
618 def test_transpose_view(self, device): argument
627 def test_transpose_inplace_view(self, device): argument
649 def test_t_view(self, device): argument
657 def test_t_inplace_view(self, device): argument
665 def test_T_view(self, device): argument
674 def test_unfold_view(self, device): argument
682 def test_squeeze_view(self, device): argument
689 def test_squeeze_inplace_view(self, device): argument
697 def test_unsqueeze_view(self, device): argument
705 def test_unsqueeze_inplace_view(self, device): argument
713 def test_as_strided_view(self, device): argument
721 def test_as_strided_inplace_view(self, device): argument
775 def test_view_view(self, device): argument
783 def test_view_as_view(self, device): argument
792 def test_contiguous_self(self, device): argument
800 def test_contiguous_nonview(self, device): argument
808 def test_reshape_view(self, device): argument
816 def test_reshape_as_view(self, device): argument
828 def test_reshape_nonview(self, device): argument
840 def test_flatten_view(self, device): argument
879 def test_flatten_nonview(self, device): argument
901 def test_basic_indexing_slice_view(self, device): argument
909 def test_basic_indexing_ellipses_view(self, device): argument
917 def test_basic_indexing_newaxis_view(self, device): argument
925 def test_advanced_indexing_nonview(self, device): argument
938 def test_advanced_indexing_assignment(self, device): argument
946 def test_chunk_view(self, device): argument
957 def test_split_view(self, device): argument
967 def test_movedim_view(self, device): argument
968 def run_test(device, op): argument
990 def test_view_copy(self, device): argument
1008 def test_view_copy_output_contiguous(self, device): argument
1013 def test_view_copy_out(self, device): argument
1034 def test_ravel(self, device): argument
1104 def test_empty_reshape(self, device): argument
1114 def test_expand(self, device): argument
1152 def test_view_empty(self, device): argument
1158 def test_reshape(self, device): argument
1192 def test_flatten(self, device): argument
1254 def test_narrow(self, device): argument
1269 def test_narrow_tensor(self, device): argument
1281 def test_t(self, device): argument
1313 def test_split(self, device): argument
1354 def test_chunk(self, device): argument
1378 def test_unsqueeze(self, device) -> None: argument
1393 def test_big_transpose(self, device): argument
1399 def test_T(self, device): argument
1408 def test_transposes(self, device, dtype): argument
1425 def test_transposes_errors(self, device, dtype): argument
1435 def test_python_types(self, device): argument
1449 def test_resize_as_preserves_strides(self, device): argument
1455 def test_memory_format_resize_as(self, device): argument
1456 def test_helper(shape, memory_format, device): argument
1467 def test_memory_format_resize_(self, device): argument
1468 def test_helper(shape, numel, memory_format, device): argument
1480 def test_transpose_invalid(self, device, dtype): argument
1493 def test_transpose_vs_numpy(self, device, dtype): argument
1539 def _test_atleast_dim(self, torch_fn, np_fn, device, dtype): argument
1567 def test_atleast(self, device, dtype): argument
1573 def _test_atleast(self, device, torch_fn): argument
1595 def test_atleast_gradient(self, device): argument
1602 def test_broadcast_tensors(self, device, dtype): argument
1614 def test_broadcast_shapes(self, device): argument
1680 def test_broadcast_to(self, device, dtype): argument
1707 def test_view(self, device): argument
1793 def test_reshape_view_semantics(self, device, dtype): argument
1810 def test_contiguous(self, device): argument
1822 def test_tensor_split_sections(self, device, dtype): argument
1855 def test_tensor_split_indices(self, device, dtype): argument
1900 def test_tensor_split_errors(self, device): argument
1965 def test_resize_all_dtypes_and_devices(self, device): argument
1972 def test_resize_as_all_dtypes_and_devices(self, device): argument
1980 def test_resize_overflow(self, device): argument
1991 def test_view_all_dtypes_and_devices(self, device): argument
1998 def test_conj_neg_view_numpy_error(self, device): argument
2021 def test_crow_col_indices(self, device): argument