Lines Matching defs:coalesced

192     def _gen_sparse(self, sparse_dim, nnz, with_size, dtype, device, coalesced):  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
437 def test_ctor_is_coalesced_with_gradcheck(self, device, dtype, coalesced): argument
521 def test_to_sparse(self, device, dtype, coalesced): 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
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
1057 def test_add_zeros(self, device, dtype, coalesced): 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
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
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
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
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
2350 def _test_log1p_tensor(self, sparse_tensor, coalesced): argument
2375 def test_log1p(self, device, dtype, coalesced): argument
2439 def test_neg_negative(self, device, dtype, coalesced): argument
2482 def _test_asin_arcsin(self, sparse_tensor, coalesced): argument
2520 def test_asin_arcsin(self, device, dtype, coalesced): argument
2566 def test_mv(self, device, dtype, coalesced): argument
2669 def test_new(self, device, dtype, coalesced): argument
3182 def test_pickle(self, device, dtype, coalesced): argument
3230 def test_div_rounding_mode(self, device, dtype, coalesced): argument
3267 def test_softmax(self, device, dtype, coalesced): argument
3472 def test_op(sparse_dims, nnz, with_size, coalesced): argument
3599 def test_sparse_matmul(self, device, dtype, coalesced): argument
3714 def test_assign(self, device, dtype, coalesced): 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
4018 def test_sum(self, device, dtype, coalesced): argument