Lines Matching defs:device

192     def _gen_sparse(self, sparse_dim, nnz, with_size, dtype, device, coalesced):  argument
226 def test_print_coalesced(self, device, dtype): argument
230 def test_print_uncoalesced(self, device, dtype): argument
233 def _test_print(self, device, dtype, coalesced): argument
288 def test_basic(self, device, dtype, coalesced): argument
324 def test_coalesce(self, device, dtype, coalesced): argument
372 def test_coalesce_reference_cycle(self, device, dtype): argument
398 def test_ctor_large_sizes(self, device, dtype): argument
412 def test_ctor_size_checks(self, device, dtype): argument
437 def test_ctor_is_coalesced_with_gradcheck(self, device, dtype, coalesced): argument
459 def test_to_dense_with_gradcheck(self, device, dtype, gradcheck): argument
521 def test_to_sparse(self, device, dtype, coalesced): argument
540 def test_sparse_bool(self, device, dtype): argument
547 def test_scalar(self, device, dtype): argument
577 def test_shared(self, device, dtype): argument
595 def test_to_dense_hybrid(self, device, dtype, gradcheck): argument
641 def test_contig(self, device, dtype): argument
722 def test_contig_hybrid(self, device, dtype): argument
810 def test_clone(self, device, dtype, coalesced): argument
829 def test_Sparse_to_Sparse_copy_(self, device, dtype, coalesced): argument
876 def test_Sparse_to_Sparse_copy_multi_gpu(self, device, dtype, coalesced): argument
906 def test_cuda_empty(self, device): argument
929 def test_transpose(self, device, dtype, coalesced): argument
951 def test_permute(self, device, dtype, coalesced, gradcheck): argument
1004 def test_coalesce_transpose_mm(self, device, dtype, coalesced): argument
1028 def test_t_empty(self, device, dtype): argument
1057 def test_add_zeros(self, device, dtype, coalesced): argument
1072 def test_add_sub_nnz(self, device, dtype): argument
1085 def test_cat(self, device, dtype, coalesced): argument
1127 def test_unsqueeze(self, device, dtype, coalesced): argument
1159 def test_select(self, device, dtype, coalesced): argument
1190 def test_select_no_type_promotion(self, device, dtype): argument
1206 def test_index_select(self, device, dtype, coalesced): argument
1230 def _test_index_select_exhaustive_index(self, sizes, dims, device, dtype, coalesced): argument
1258 def test_index_select_exhaustive_index_small(self, device, dtype, coalesced): argument
1264 def test_index_select_exhaustive_index_large(self, device, dtype, coalesced): argument
1270 def test_index_select_empty_and_non_contiguous_index(self, device, dtype, coalesced): argument
1302 def test_index_select_parallelization(self, device, dtype, coalesced): argument
1333 def test_mm(self, device, dtype, coalesced): argument
1367 def test_bmm(self, device, dtype, coalesced): argument
1423 def test_bmm_deterministic(self, device, dtype, coalesced): argument
1462 def test_bmm_windows_error(self, device, dtype): argument
1473 def test_saddmm(self, device, dtype, coalesced): argument
1506 def test_sspaddmm(self, device, dtype, coalesced): argument
1557 def test_sparse_addmm(self, device, dtype, coalesced): argument
1598 def test_sparse_mm(self, device, dtype, coalesced): argument
1621 def test_sparse_mul(self, device, dtype, coalesced, gradcheck): argument
1642 def test_dsmm(self, device, dtype, coalesced): argument
1661 def test_hsmm(self, device, dtype, coalesced): argument
1680 def test_spadd(self, device, dtype, coalesced): argument
1752 def test_sparse_add_out_bfloat16(self, device, dtype, coalesced): argument
1767 def test_norm(self, device, dtype, coalesced): argument
1796 def test_sparse_sum(self, device, dtype, coalesced): argument
1858 def _test_basic_ops_shape(self, nnz_x1, nnz_x2, shape_i, shape_v, dtype, device, coalesced): argument
1935 def test_basic_ops(self, device, dtype, coalesced): argument
1966 def test_add_dense_sparse_mismatch(self, device, dtype): argument
1982 def test_add_noncontiguous(self, device, dtype): argument
1991 def _test_sparse_mask_shape(self, nnz_x1, nnz_x2, shape_i, shape_v, dtype, device, coalesced): argument
2005 def test_sparse_mask(self, device, dtype, coalesced): argument
2074 def test_sparse_mask_hybrid(self, device, dtype, coalesced): argument
2136 def test_sparse_mask_backward(self, device, dtype): argument
2169 def test_zeros(self, device, dtype, coalesced): argument
2194 def test_zeros_like(self, device, dtype, coalesced): argument
2247 def _test_empty_like(self, sparse_tensor, dtype, device, coalesced): argument
2277 def test_empty_like(self, device, dtype, coalesced): argument
2333 def test_narrow(self, device, dtype, coalesced): argument
2375 def test_log1p(self, device, dtype, coalesced): argument
2439 def test_neg_negative(self, device, dtype, coalesced): argument
2520 def test_asin_arcsin(self, device, dtype, coalesced): argument
2566 def test_mv(self, device, dtype, coalesced): argument
2592 def test_sparse_add_coalesce(self, device, dtype): argument
2610 def test_storage_not_null(self, device): argument
2643 def _test_new_device(self, size, device=torch.cuda): argument
2669 def test_new(self, device, dtype, coalesced): argument
2685 def test_factory(self, device, dtype): argument
2725 def test_factory_size_check(self, device, dtype): argument
2765 def test_factory_empty_indices(self, device): argument
2779 def test_factory_nnz(self, device, dtype): argument
2793 def test_factory_nnz_zero(self, device, dtype): argument
2814 def test_factory_dense_dim(self, device, dtype): argument
2829 def test_factory_type_inference(self, device, dtype): argument
2845 def test_factory_device_type_inference(self, device): argument
2871 def test_factory_copy(self, device): argument
2933 def test_legacy_new_device(self, device): argument
2951 def test_legacy_new(self, device): argument
2964 def test_dtypes(self, device): argument
2970 def _test_empty_full(self, device, dtype, requires_grad): argument
3002 def test_empty_full(self, device, dtype, requires_grad): argument
3011 def test_is_sparse(self, device): argument
3021 def test_resize_as(self, device): argument
3032 def _test_resize_shape(self, x_i, x_v, x_size, y_i, y_v, y_size, dtype, device): argument
3054 def test_resize(self, device, dtype): argument
3118 def test_is_nonzero(self, device): argument
3139 def test_change_tensor_metadata(self, device, dtype): argument
3182 def test_pickle(self, device, dtype, coalesced): argument
3212 def test_any(self, device): argument
3220 def test_isnan(self, device): argument
3230 def test_div_rounding_mode(self, device, dtype, coalesced): argument
3249 def test_div_by_sparse_error(self, device): argument
3254 def test_floor_divide_by_sparse_error(self, device): argument
3261 def test_sparse_to_numpy(self, device): argument
3267 def test_softmax(self, device, dtype, coalesced): argument
3566 def _check_zero_nnz_softmax_op(self, func, ndim, device, dtype): argument
3579 def test_softmax_zero_nnz(self, device, dtype): argument
3585 def test_log_softmax_zero_nnz(self, device, dtype): argument
3599 def test_sparse_matmul(self, device, dtype, coalesced): argument
3714 def test_assign(self, device, dtype, coalesced): argument
3722 def test_full_broadcast_to(self, device, dtype): argument
3753 def test_sparse_broadcast_to(self, device, dtype, coalesced): argument
3766 def _test_mul_skips(self, device, dtype, coalesced): argument
3784 def test_sparse_sparse_mul(self, device, dtype, coalesced): argument
3834 def test_sparse_dense_mul(self, device, dtype, coalesced): argument
3922 def test_sparse_spdiags(self, device, dtype): argument
4018 def test_sum(self, device, dtype, coalesced): argument
4097 def test_sparse_consistency(self, device, dtype, op): argument
4108 def test_out(self, device, dtype, op): argument
4122 def test_inplace(self, device, dtype, op): argument
4140 def test_sparse_zero_dims(self, device, dtype, op): argument
4150 def test_sparse_zeros(self, device, dtype, op): argument
4162 def test_sparse_fn_grad(self, device, dtype, op): argument
4191 def test_future_empty_dim(self, device, dtype, op): argument
4714 def test_constructor_autograd(self, device, layout): argument
4784 def test_to_dense(self, from_layout, device, dtype, index_dtype): argument
4798 def test_gradcheck_to_dense(self, from_layout, device, dtype, index_dtype, gradcheck): argument
4813 def test_to_sparse(self, from_layout, to_layout, device, dtype, index_dtype): argument
4987 def test_reductions(self, layout, device, dtype, op): argument
5007 def test_reductions_backward(self, layout, device, dtype, op): argument
5039 def test_unsupported_backend_error_message(self, mth, layout, device): argument
5076 def test_gradcheck_mm(self, layout, dtype, device, masked, fast_mode): argument
5121 def test_binary_operation(self, layout, device, dtype, op): argument
5156 def test_to_sparse_identity(self, device, layout, dtype): argument
5176 def test_like_fns(self, layout, device, dtype, op): argument
5231 def test_sparse_mask(self, mask_layout, device, dtype): argument
5275 def test_as_sparse_gradcheck(self, layout, device, masked, fast_mode): argument
5315 def test_dataloader(self, device, layout, dtype): argument
5344 def test_constructor_pin_memory(self, device, layout): argument
5374 def test_method_pin_memory(self, device, layout): argument
5424 def test_constructor_pinned_memory(self, device, layout): argument
5455 def test_constructor_mismatched_pinned_memory(self, device, layout): argument