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

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

__init__(self, method_name='runTest') (defined in common_utils.TestCase)common_utils.TestCase
a (defined in test_jit.TestJit)test_jit.TestJit
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
b (defined in test_jit.TestJit)test_jit.TestJit
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
do_trace_size(self, requires_grad) (defined in test_jit.TestJit)test_jit.TestJit
emitModuleHook(self, module) (defined in test_jit.JitTestCase)test_jit.JitTestCase
features (defined in test_jit.TestJit)test_jit.TestJit
foo (defined in test_jit.TestJit)test_jit.TestJit
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
mod (defined in test_jit.TestJit)test_jit.TestJit
module (defined in test_jit.TestJit)test_jit.TestJit
num_layers (defined in test_jit.TestJit)test_jit.TestJit
p0 (defined in test_jit.TestJit)test_jit.TestJit
precision (defined in common_utils.TestCase)common_utils.TestCasestatic
rnn (defined in test_jit.TestJit)test_jit.TestJit
run_ge_tests(self, optimize, use_cuda) (defined in test_jit.TestJit)test_jit.TestJit
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
submodule (defined in test_jit.TestJit)test_jit.TestJit
tearDown(self) (defined in test_jit.JitTestCase)test_jit.JitTestCase
test_alexnet(self) (defined in test_jit.TestJit)test_jit.TestJit
test_arg_configurations(self)test_jit.TestJit
test_batchnorm(self) (defined in test_jit.TestJit)test_jit.TestJit
test_big(self) (defined in test_jit.TestJit)test_jit.TestJit
test_canonicalize_tensor_iterator(self) (defined in test_jit.TestJit)test_jit.TestJit
test_constant(self) (defined in test_jit.TestJit)test_jit.TestJit
test_constant_prop_if_constant(self) (defined in test_jit.TestJit)test_jit.TestJit
test_constant_prop_if_inline(self) (defined in test_jit.TestJit)test_jit.TestJit
test_constant_prop_loop_constant(self) (defined in test_jit.TestJit)test_jit.TestJit
test_constant_prop_nested(self) (defined in test_jit.TestJit)test_jit.TestJit
test_constant_prop_none(self) (defined in test_jit.TestJit)test_jit.TestJit
test_constant_prop_print(self) (defined in test_jit.TestJit)test_jit.TestJit
test_constant_prop_rand(self) (defined in test_jit.TestJit)test_jit.TestJit
test_constant_prop_remove_output(self) (defined in test_jit.TestJit)test_jit.TestJit
test_constant_prop_simple(self) (defined in test_jit.TestJit)test_jit.TestJit
test_conv(self) (defined in test_jit.TestJit)test_jit.TestJit
test_cpp_cuda(self) (defined in test_jit.TestJit)test_jit.TestJit
test_cse(self) (defined in test_jit.TestJit)test_jit.TestJit
test_cu_escaped_number(self) (defined in test_jit.TestJit)test_jit.TestJit
test_cuda_export_restore(self) (defined in test_jit.TestJit)test_jit.TestJit
test_decompose_addmm(self) (defined in test_jit.TestJit)test_jit.TestJit
test_diff_subgraph_clones_constants(self) (defined in test_jit.TestJit)test_jit.TestJit
test_disabled(self) (defined in test_jit.TestJit)test_jit.TestJit
test_dropout(self) (defined in test_jit.TestJit)test_jit.TestJit
test_dropout_cuda(self) (defined in test_jit.TestJit)test_jit.TestJit
test_einsum(self) (defined in test_jit.TestJit)test_jit.TestJit
test_export_batchnorm(self) (defined in test_jit.TestJit)test_jit.TestJit
test_export_dropout(self) (defined in test_jit.TestJit)test_jit.TestJit
test_export_lstm(self) (defined in test_jit.TestJit)test_jit.TestJit
test_export_no_reorder(self) (defined in test_jit.TestJit)test_jit.TestJit
test_export_rnn(self) (defined in test_jit.TestJit)test_jit.TestJit
test_export_tensoroption_to(self) (defined in test_jit.TestJit)test_jit.TestJit
test_flags(self) (defined in test_jit.TestJit)test_jit.TestJit
test_function_default_values(self) (defined in test_jit.TestJit)test_jit.TestJit
test_ge(self) (defined in test_jit.TestJit)test_jit.TestJit
test_ge_cuda(self) (defined in test_jit.TestJit)test_jit.TestJit
test_ge_optimized(self) (defined in test_jit.TestJit)test_jit.TestJit
test_ge_unoptimized(self) (defined in test_jit.TestJit)test_jit.TestJit
test_import_method(self) (defined in test_jit.TestJit)test_jit.TestJit
test_index(self) (defined in test_jit.TestJit)test_jit.TestJit
test_index_constant(self) (defined in test_jit.TestJit)test_jit.TestJit
test_index_put(self) (defined in test_jit.TestJit)test_jit.TestJit
test_inplace_check(self) (defined in test_jit.TestJit)test_jit.TestJit
test_inplace_copy(self) (defined in test_jit.TestJit)test_jit.TestJit
test_inplace_flags(self) (defined in test_jit.TestJit)test_jit.TestJit
test_inplace_transplant(self) (defined in test_jit.TestJit)test_jit.TestJit
test_input_flatten(self)test_jit.TestJit
test_legacy_constructors(self) (defined in test_jit.TestJit)test_jit.TestJit
test_legacy_fail(self) (defined in test_jit.TestJit)test_jit.TestJit
test_model_save_error(self) (defined in test_jit.TestJit)test_jit.TestJit
test_module_default_values(self) (defined in test_jit.TestJit)test_jit.TestJit
test_nested_inplace(self) (defined in test_jit.TestJit)test_jit.TestJit
test_no_erroneous_warnings(self) (defined in test_jit.TestJit)test_jit.TestJit
test_onnx_transpose_incomplete_tensor_type(self) (defined in test_jit.TestJit)test_jit.TestJit
test_output_unflatten(self)test_jit.TestJit
test_peephole(self) (defined in test_jit.TestJit)test_jit.TestJit
test_peephole_cuda(self) (defined in test_jit.TestJit)test_jit.TestJit
test_peephole_dynamic(self) (defined in test_jit.TestJit)test_jit.TestJit
test_pretty_printer(self) (defined in test_jit.TestJit)test_jit.TestJit
test_python_bindings(self) (defined in test_jit.TestJit)test_jit.TestJit
test_python_function(self) (defined in test_jit.TestJit)test_jit.TestJit
test_python_function_tup(self) (defined in test_jit.TestJit)test_jit.TestJit
test_python_ir(self) (defined in test_jit.TestJit)test_jit.TestJit
test_recursive_cse(self) (defined in test_jit.TestJit)test_jit.TestJit
test_repeated_input(self) (defined in test_jit.TestJit)test_jit.TestJit
test_repeated_output(self) (defined in test_jit.TestJit)test_jit.TestJit
test_restore_device(self) (defined in test_jit.TestJit)test_jit.TestJit
test_restore_device_cuda(self) (defined in test_jit.TestJit)test_jit.TestJit
test_restore_shared_storage_on_cuda(self) (defined in test_jit.TestJit)test_jit.TestJit
test_scopes(self) (defined in test_jit.TestJit)test_jit.TestJit
test_scopes_identity_node(self) (defined in test_jit.TestJit)test_jit.TestJit
test_scopes_intermediate_node(self) (defined in test_jit.TestJit)test_jit.TestJit
test_shape_analysis_broadcast(self) (defined in test_jit.TestJit)test_jit.TestJit
test_shared_param(self) (defined in test_jit.TestJit)test_jit.TestJit
test_simple(self) (defined in test_jit.TestJit)test_jit.TestJit
test_single_tuple_trace(self) (defined in test_jit.TestJit)test_jit.TestJit
test_sparse_tensors_error(self) (defined in test_jit.TestJit)test_jit.TestJit
test_torch_load_error(self) (defined in test_jit.TestJit)test_jit.TestJit
test_trace_annotation(self) (defined in test_jit.TestJit)test_jit.TestJit
test_trace_c10_ops(self) (defined in test_jit.TestJit)test_jit.TestJit
test_trace_casts(self) (defined in test_jit.TestJit)test_jit.TestJit
test_trace_detach(self) (defined in test_jit.TestJit)test_jit.TestJit
test_trace_detach_inplace(self) (defined in test_jit.TestJit)test_jit.TestJit
test_trace_detach_onnx_erase(self) (defined in test_jit.TestJit)test_jit.TestJit
test_trace_dict_input(self) (defined in test_jit.TestJit)test_jit.TestJit
test_trace_indexed_assignment(self) (defined in test_jit.TestJit)test_jit.TestJit
test_trace_random(self) (defined in test_jit.TestJit)test_jit.TestJit
test_trace_records_names(self) (defined in test_jit.TestJit)test_jit.TestJit
test_trace_script(self) (defined in test_jit.TestJit)test_jit.TestJit
test_trace_size(self) (defined in test_jit.TestJit)test_jit.TestJit
test_trace_size_with_grad(self) (defined in test_jit.TestJit)test_jit.TestJit
test_trace_slice_expr_complete_type(self) (defined in test_jit.TestJit)test_jit.TestJit
test_trace_slice_full_dim(self) (defined in test_jit.TestJit)test_jit.TestJit
test_trace_tensor_factory(self) (defined in test_jit.TestJit)test_jit.TestJit
test_trace_tuple(self) (defined in test_jit.TestJit)test_jit.TestJit
test_trace_variable_instantiation(self) (defined in test_jit.TestJit)test_jit.TestJit
test_trace_warn(self) (defined in test_jit.TestJit)test_jit.TestJit
test_traced_module_cuda(self) (defined in test_jit.TestJit)test_jit.TestJit
test_train_eval(self) (defined in test_jit.TestJit)test_jit.TestJit
test_tuple_specialization(self) (defined in test_jit.TestJit)test_jit.TestJit
test_typeas_trace_check(self) (defined in test_jit.TestJit)test_jit.TestJit
test_verify(self) (defined in test_jit.TestJit)test_jit.TestJit
test_warnings(self) (defined in test_jit.TestJit)test_jit.TestJit
weight (defined in test_jit.TestJit)test_jit.TestJit
wrap_with_cuda_memory_check(self, method) (defined in common_utils.TestCase)common_utils.TestCase