 |
Caffe2 - Python API
A deep learning, cross platform ML framework
|
|
Caffe2 - Python API
A deep learning, cross platform ML framework
|
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 |
Definition at line 36 of file test_sparse.py.
| 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.
1.8.11
