Searched refs:check_sparse_tensor_invariants (Results 1 – 5 of 5) sorted by relevance
4509 @torch.sparse.check_sparse_tensor_invariants(enable=False)4553 self.assertFalse(torch.sparse.check_sparse_tensor_invariants.is_enabled())4556 with torch.sparse.check_sparse_tensor_invariants():4557 self.assertTrue(torch.sparse.check_sparse_tensor_invariants.is_enabled())4561 self.assertFalse(torch.sparse.check_sparse_tensor_invariants.is_enabled())4581 self.assertFalse(torch.sparse.check_sparse_tensor_invariants.is_enabled())4585 with torch.sparse.check_sparse_tensor_invariants():4586 self.assertTrue(torch.sparse.check_sparse_tensor_invariants.is_enabled())4608 self.assertTrue(torch.sparse.check_sparse_tensor_invariants.is_enabled())4612 self.assertFalse(torch.sparse.check_sparse_tensor_invariants.is_enabled())[all …]
316 with torch.sparse.check_sparse_tensor_invariants(enable=False):1515 @torch.sparse.check_sparse_tensor_invariants(enable=False)
437 class check_sparse_tensor_invariants: class
471 globally using :class:`torch.sparse.check_sparse_tensor_invariants`687 globally using :class:`torch.sparse.check_sparse_tensor_invariants`1335 sparse.check_sparse_tensor_invariants
3122 self._check_invariants = torch.sparse.check_sparse_tensor_invariants.is_enabled()3130 torch.sparse.check_sparse_tensor_invariants.enable()3145 torch.sparse.check_sparse_tensor_invariants.enable()3147 torch.sparse.check_sparse_tensor_invariants.disable()