 |
Caffe2 - Python API
A deep learning, cross platform ML framework
|
|
Caffe2 - Python API
A deep learning, cross platform ML framework
|
Public Member Functions | |
| def | rand_batch (self, dims) |
| def | test_create_batchtensor (self) |
| def | test_batch_elementwise_unary (self) |
| def | test_batch_elementwise_binary (self) |
| def | test_batch_mm (self) |
| def | test_batch_matmul (self) |
| def | test_batch_select (self) |
| def | test_batch_index_select (self) |
| def | test_batch_where (self) |
| def | test_batch_argmax (self) |
| def | test_batch_topk (self) |
| def | test_batch_softmax (self) |
| def | test_batch_view (self) |
| def | test_batch_cat (self) |
| def | test_batch_sum (self) |
| def | test_if_else (self) |
| def | test_if_else_with_scalar (self) |
| def | test_if_noelse (self) |
| def | test_if_noelse_with_scalar (self) |
| def | test_while (self) |
| def | test_for (self) |
| def | test_lstm (self) |
| def | test_greedy_search (self) |
| def | test_beam_search (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) |
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 2426 of file test_jit.py.
1.8.11
