Lines Matching defs:device

35     def test_index(self, device):  argument
182 def test_advancedindex(self, device, dtype): argument
798 def test_advancedindex_big(self, device): argument
806 def test_set_item_to_scalar_tensor(self, device): argument
816 def test_single_int(self, device): argument
820 def test_multiple_int(self, device): argument
825 def test_none(self, device): argument
832 def test_step(self, device): argument
840 def test_step_assignment(self, device): argument
846 def test_bool_indices(self, device): argument
869 def test_bool_indices_accumulate(self, device): argument
875 def test_multiple_bool_indices(self, device): argument
882 def test_byte_mask(self, device): argument
894 def test_byte_mask_accumulate(self, device): argument
907 def test_index_put_accumulate_large_tensor(self, device): argument
945 def test_index_put_accumulate_expanded_values(self, device): argument
984 def test_index_put_accumulate_non_contiguous(self, device): argument
1004 def test_index_put_accumulate_with_optional_tensors(self, device): argument
1032 def test_index_put_accumulate_duplicate_indices(self, device): argument
1052 def test_index_ind_dtype(self, device): argument
1077 def test_index_put_accumulate_empty(self, device): argument
1083 def test_multiple_byte_mask(self, device): argument
1093 def test_byte_mask2d(self, device): argument
1101 def test_jit_indexing(self, device): argument
1123 def test_int_indices(self, device): argument
1145 def test_index_put_src_datatype(self, device, dtype): argument
1155 def test_index_src_datatype(self, device, dtype): argument
1164 def test_int_indices2d(self, device): argument
1171 def test_int_indices_broadcast(self, device): argument
1179 def test_empty_index(self, device): argument
1193 def test_empty_ndim_index(self, device): argument
1212 def test_empty_ndim_index_bool(self, device): argument
1218 def test_empty_slice(self, device): argument
1227 def test_index_getitem_copy_bools_slices(self, device): argument
1241 def test_index_setitem_bools_slices(self, device): argument
1268 def test_index_scalar_with_bool_mask(self, device): argument
1279 def test_setitem_expansion_error(self, device): argument
1290 def test_getitem_scalars(self, device): argument
1314 def test_setitem_scalars(self, device): argument
1338 def test_basic_advanced_combined(self, device): argument
1354 def test_int_assignment(self, device): argument
1363 def test_byte_tensor_assignment(self, device): argument
1377 def test_variable_slicing(self, device): argument
1383 def test_ellipsis_tensor(self, device): argument
1389 def test_unravel_index_errors(self, device): argument
1416 def test_invalid_index(self, device): argument
1420 def test_out_of_bound_index(self, device): argument
1443 def test_zero_dim_index(self, device): argument
1454 def test_invalid_device(self, device): argument
1466 def test_cpu_indices(self, device): argument
1478 def test_take_along_dim(self, device, dtype): argument
1512 def test_take_along_dim_invalid(self, device, dtype): argument
1543 def test_gather_take_along_dim_cross_device(self, device, dtype): argument
1572 def test_cuda_broadcast_index_use_deterministic_algorithms(self, device): argument
1612 def test_index_limits(self, device): argument
1657 def test_index_no_floats(self, device): argument
1683 def test_none_index(self, device): argument
1688 def test_empty_tuple_index(self, device): argument
1694 def test_empty_fancy_index(self, device): argument
1705 def test_ellipsis_index(self, device): argument
1727 def test_single_int_index(self, device): argument
1739 def test_single_bool_index(self, device): argument
1746 def test_boolean_shape_mismatch(self, device): argument
1759 def test_boolean_indexing_onedim(self, device): argument
1771 def test_boolean_assignment_value_mismatch(self, device): argument
1783 def test_boolean_indexing_twodim(self, device): argument
1799 def test_boolean_indexing_weirdness(self, device): argument
1808 def test_boolean_indexing_weirdness_tensors(self, device): argument
1819 def test_boolean_indexing_alldims(self, device): argument
1825 def test_boolean_list_indexing(self, device): argument
1836 def test_everything_returns_views(self, device): argument
1844 def test_broaderrors_indexing(self, device): argument
1853 def test_trivial_fancy_out_of_bounds(self, device): argument
1866 def test_index_is_larger(self, device): argument
1873 def test_broadcast_subspace(self, device): argument
1881 def test_truncate_leading_1s(self, device): argument