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

 is_cuda
 
 is_uncoalesced
 
 device
 
 value_dtype
 
 index_tensor
 
 value_empty
 
 value_tensor
 
 sparse_empty
 
 sparse_tensor
 
 legacy_sparse_tensor
 

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 test_sparse.py.

Member Function Documentation

def test_sparse.TestSparse.assert_uncoalesced (   self,
  x 
)
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 test_sparse.py.

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

Definition at line 90 of file test_sparse.py.


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