Lines Matching full:is_coalesced
208 assert not x.is_coalesced()
296 self.assertEqual(x.coalesce()._nnz(), nnz if x.is_coalesced() else nnz // 2)
329 self.assertTrue(tc.is_coalesced())
358 if t.is_coalesced():
391 self.assertTrue(S2.is_coalesced())
440 self.assertEqual(t.is_coalesced(), coalesced)
442 def func(indices, values, shape, is_coalesced): argument
443 …= torch.sparse_coo_tensor(indices, values, shape, check_invariants=True, is_coalesced=is_coalesced)
444 self.assertEqual(s.is_coalesced(), is_coalesced)
453 … "cannot set is_coalesced to true if indices correspond to uncoalesced COO tensor"):
553 self.assertTrue(a_coalesced.is_coalesced())
563 self.assertTrue(a_coalesced.is_coalesced())
572 self.assertTrue(a_coalesced.is_coalesced())
814 self.assertFalse(x.is_coalesced())
816 self.assertFalse(y.is_coalesced())
818 self.assertTrue(x.is_coalesced())
820 self.assertTrue(y.is_coalesced())
984 self.assertEqual(s_permuted.is_coalesced(), s.is_coalesced())
986 self.assertFalse(s_permuted.is_coalesced())
1010 self.assertTrue(x_coalesced.is_coalesced())
1014 self.assertEqual(x_coalesced_t.is_coalesced(), di * nnz == 0)
1535 self.assertTrue(weight.is_coalesced())
1918 self.assertEqual(x1.is_coalesced(), coalesced)
1921 self.assertEqual(x1.is_coalesced(), coalesced)
1922 self.assertTrue(y.is_coalesced())
1924 if not x1.is_coalesced():
2027 self.assertTrue(sparse.is_coalesced())
2044 self.assertTrue(sparse.is_coalesced())
2070 self.assertEqual(rhs.is_coalesced(), res_all_sparse.is_coalesced())
2099 self.assertTrue(sparse.is_coalesced())
2116 self.assertTrue(sparse.is_coalesced())
2599 self.assertFalse(z._indices().numel() != 2 and z.is_coalesced())
2607 self.assertFalse(z._indices().numel() != 2 and z.is_coalesced())
3650 self.assertTrue(r2.is_coalesced() and is_coalesced_indices(r2))
3741 if s_res.is_coalesced():
3742 # ensure that is_coalesced is estimated correctly
4008 self.assertTrue(torch.sparse_coo_tensor([[], []], [], (2, 2)).is_coalesced())
4010 self.assertTrue(torch.sparse_coo_tensor([[0], [0]], [1], (2, 2)).is_coalesced())
4012 self.assertFalse(torch.sparse_coo_tensor([[0, 0], [0, 0]], [1, 2], (2, 2)).is_coalesced())
4014 self.assertFalse(torch.sparse_coo_tensor([[0, 1], [0, 1]], [1, 2], (2, 2)).is_coalesced())
4256 self.assertEqual(r.is_coalesced(), True)
4259 self.assertEqual(r.is_coalesced(), False)
4925 self.assertEqual(t.is_coalesced(), r.is_coalesced())
4927 # Check r is truly coalesced when r.is_coalesced == True
4928 if r.is_coalesced():
5029 for mth in [torch.Tensor.is_coalesced,
5047 is_coalesced=({torch.sparse_coo},
5048 …"is_coalesced expected sparse coordinate tensor layout but got (Sparse(Csr|Csc|Bsr|Bsc)|Strided)"),
5247 is_coalesced=mask.is_coalesced())