Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
test_sparse.TestSparse Class Reference
Inheritance diagram for test_sparse.TestSparse:
common_utils.TestCase expecttest.TestCase test_sparse.TestCudaSparse test_sparse.TestUncoalescedSparse test_sparse.TestCudaUncoalescedSparse

Public Member Functions

def setUp (self)
def assert_uncoalesced (self, x)
def randn (self, args, kwargs)
def test_print (self)
def test_basic (self)
def test_coalecce (self)
def test_ctor_size_checks (self)
def test_to_dense (self)
def test_to_sparse (self)
def test_scalar (self)
def test_shared (self)
def test_to_dense_hybrid (self)
def test_contig (self)
def test_contig_hybrid (self)
def test_clone (self)
def test_Sparse_to_Sparse_copy_ (self)
def test_Sparse_to_Sparse_copy_multi_gpu (self)
def test_cuda_empty (self)
def test_transpose (self)
def test_coalesce_transpose_mm (self)
def test_t_empty (self)
def test_add_zeros (self)
def test_cat (self)
def test_unsqueeze (self)
def test_mm (self)
def test_saddmm (self)
def test_sparse_addmm (self)
def test_sparse_mm (self)
def test_dsmm (self)
def test_hsmm (self)
def test_spadd (self)
def test_spadd_hybrid (self)
def test_norm (self)
def test_sparse_sum (self)
def test_basic_ops (self)
def test_basic_ops_hybrid (self)
def test_add_dense_sparse_mismatch (self)
def test_sparse_mask (self)
def test_sparse_mask_hybrid (self)
def test_zeros (self)
def test_zeros_like (self)
def test_narrow (self)
def test_log1p (self)
def test_sparse_add_coalesce (self)
def test_storage_not_null (self)
def test_same_gpu (self)
def test_new_device_single_gpu (self)
def test_new_device_multi_gpu (self)
def test_new (self)
def test_factory (self)
def test_factory_size_check (self)
def test_factory_default (self)
def test_factory_empty_indices (self)
def test_factory_nnz (self)
def test_factory_nnz_zero (self)
def test_factory_dense_dim (self)
def test_factory_type_inference (self)
def test_factory_device_type_inference (self)
def test_factory_copy (self)
def test_constructor_device_legacy (self)
def test_dtypes (self)
def test_empty_full (self)
def test_is_sparse (self)
def test_resize_as (self)
def test_resize (self)
def test_is_nonzero (self)
def test_allow_tensor_metadata_change (self)
- Public Member Functions inherited from common_utils.TestCase
def __init__ (self, method_name='runTest')
def assertLeaksNoCudaTensors (self, name=None)
def wrap_with_cuda_memory_check (self, method)
def setUp (self)
def assertTensorsSlowEqual (self, x, y, prec=None, message='')
def genSparseTensor (self, size, sparse_dim, nnz, is_uncoalesced, device='cpu')
def safeToDense (self, t)
def safeCoalesce (self, t)
def assertEqual (self, x, y, prec=None, message='', allow_inf=False)
def assertAlmostEqual (self, x, y, places=None, msg=None, delta=None, allow_inf=None)
def assertNotEqual (self, x, y, prec=None, message='')
def assertObjectIn (self, obj, iterable)
def assertExpectedRaises (self, exc_type, callable, args, kwargs)
def assertWarns (self, callable, msg='')
def assertWarnsRegex (self, callable, regex, msg='')
def assertExpected (self, s, subname=None)
- Public Member Functions inherited from expecttest.TestCase
def assertExpectedInline (self, actual, expect, skip=0)

Public Attributes


Additional Inherited Members

- Static Public Attributes inherited from common_utils.TestCase
int precision = 1
 maxDiff = None
 assertRegex = unittest.TestCase.assertRegexpMatches
 assertRaisesRegex = unittest.TestCase.assertRaisesRegexp
- Static Public Attributes inherited from expecttest.TestCase
bool longMessage = True

Detailed Description

Definition at line 36 of file

Member Function Documentation

def test_sparse.TestSparse.assert_uncoalesced (   self,
Test if a CPU tensor is uncoalesced.  This is used to ensure
correctness of the uncoalesced tensor generation algorithm.

Definition at line 76 of file

def test_sparse.TestSparse.randn (   self,
Variant of torch.randn that also works in the TEST_CUDA case.

Definition at line 90 of file

The documentation for this class was generated from the following file: