Caffe2 - Python API
A deep learning, cross platform ML framework
test_jit.TestPytorchExportModes Member List

This is the complete list of members for test_jit.TestPytorchExportModes, including all inherited members.

__init__(self, method_name='runTest') (defined in common_utils.TestCase)common_utils.TestCase
assertAlmostEqual(self, x, y, places=None, msg=None, delta=None, allow_inf=None) (defined in common_utils.TestCase)common_utils.TestCase
assertEqual(self, x, y, prec=None, message='', allow_inf=False) (defined in common_utils.TestCase)common_utils.TestCase
assertExpected(self, s, subname=None) (defined in common_utils.TestCase)common_utils.TestCase
assertExpectedGraph(self, trace, args, kwargs) (defined in test_jit.JitTestCase)test_jit.JitTestCase
assertExpectedInline(self, actual, expect, skip=0) (defined in expecttest.TestCase)expecttest.TestCase
assertExpectedONNXGraph(self, trace, args, kwargs) (defined in test_jit.JitTestCase)test_jit.JitTestCase
assertExpectedRaises(self, exc_type, callable, args, kwargs) (defined in common_utils.TestCase)common_utils.TestCase
assertExportImport(self, trace, inputs) (defined in test_jit.JitTestCase)test_jit.JitTestCase
assertExportImportModule(self, m, inputs) (defined in test_jit.JitTestCase)test_jit.JitTestCase
assertGraphContains(self, graph, kind) (defined in test_jit.JitTestCase)test_jit.JitTestCase
assertGraphContainsExactly(self, graph, kind, num_kind_nodes, consider_subgraphs=False) (defined in test_jit.JitTestCase)test_jit.JitTestCase
assertLeaksNoCudaTensors(self, name=None) (defined in common_utils.TestCase)common_utils.TestCase
assertNotEqual(self, x, y, prec=None, message='') (defined in common_utils.TestCase)common_utils.TestCase
assertObjectIn(self, obj, iterable) (defined in common_utils.TestCase)common_utils.TestCase
assertRaisesRegex (defined in common_utils.TestCase)common_utils.TestCasestatic
assertRegex (defined in common_utils.TestCase)common_utils.TestCasestatic
assertTensorsSlowEqual(self, x, y, prec=None, message='') (defined in common_utils.TestCase)common_utils.TestCase
assertWarns(self, callable, msg='') (defined in common_utils.TestCase)common_utils.TestCase
assertWarnsRegex(self, callable, regex, msg='') (defined in common_utils.TestCase)common_utils.TestCase
checkScript(self, script, inputs, optimize=True, outputs=None, name='func', capture_output=False, frames_up=1, check_expected=False) (defined in test_jit.JitTestCase)test_jit.JitTestCase
checkTrace(self, func, reference_tensors, input_tensors=None, optimize=True, drop=None, allow_unused=False, verbose=False, inputs_require_grads=True, check_tolerance=1e-5, export_import=True, _force_outplace=False) (defined in test_jit.JitTestCase)test_jit.JitTestCase
createScriptModuleFromGraph(self, trace) (defined in test_jit.JitTestCase)test_jit.JitTestCase
disableModuleHook(self) (defined in test_jit.JitTestCase)test_jit.JitTestCase
emitModuleHook(self, module) (defined in test_jit.JitTestCase)test_jit.JitTestCase
genSparseTensor(self, size, sparse_dim, nnz, is_uncoalesced, device='cpu') (defined in common_utils.TestCase)common_utils.TestCase
getExportImportCopy(self, m, also_test_file=True, map_location=None) (defined in test_jit.JitTestCase)test_jit.JitTestCase
getExportImportCopyWithPacking(self, m, also_test_file=True, map_location=None) (defined in test_jit.JitTestCase)test_jit.JitTestCase
longMessage (defined in expecttest.TestCase)expecttest.TestCasestatic
maxDiff (defined in common_utils.TestCase)common_utils.TestCasestatic
precision (defined in common_utils.TestCase)common_utils.TestCasestatic
run_pass(self, name, trace) (defined in test_jit.JitTestCase)test_jit.JitTestCase
runAndSaveRNG(self, func, inputs, kwargs=None) (defined in test_jit.JitTestCase)test_jit.JitTestCase
safeCoalesce(self, t) (defined in common_utils.TestCase)common_utils.TestCase
safeToDense(self, t) (defined in common_utils.TestCase)common_utils.TestCase
setUp(self) (defined in test_jit.JitTestCase)test_jit.JitTestCase
tearDown(self) (defined in test_jit.JitTestCase)test_jit.JitTestCase
test_aten_fallback(self) (defined in test_jit.TestPytorchExportModes)test_jit.TestPytorchExportModes
test_compressed_zipfile(self) (defined in test_jit.TestPytorchExportModes)test_jit.TestPytorchExportModes
test_directory(self) (defined in test_jit.TestPytorchExportModes)test_jit.TestPytorchExportModes
test_onnx_aten(self) (defined in test_jit.TestPytorchExportModes)test_jit.TestPytorchExportModes
test_onnx_multiple_return(self) (defined in test_jit.TestPytorchExportModes)test_jit.TestPytorchExportModes
test_protobuf(self) (defined in test_jit.TestPytorchExportModes)test_jit.TestPytorchExportModes
test_zipfile(self) (defined in test_jit.TestPytorchExportModes)test_jit.TestPytorchExportModes
wrap_with_cuda_memory_check(self, method) (defined in common_utils.TestCase)common_utils.TestCase