| __init__(self, method_name='runTest') (defined in common_utils.TestCase) | common_utils.TestCase | |
| affine1 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| affine2 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| 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.TestCase | static |
| assertRegex (defined in common_utils.TestCase) | common_utils.TestCase | static |
| 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 | |
| config (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| conv1 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| conv2 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| conv2d (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| conv3 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| conv4 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| createScriptModuleFromGraph(self, trace) (defined in test_jit.JitTestCase) | test_jit.JitTestCase | |
| deconv1 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| deconv2 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| deconv3 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| disableModuleHook(self) (defined in test_jit.JitTestCase) | test_jit.JitTestCase | |
| dropout (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| embed (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| emitModuleHook(self, module) (defined in test_jit.JitTestCase) | test_jit.JitTestCase | |
| encoder (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| fc1 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| fc21 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| fc22 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| fc3 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| fc4 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| 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 | |
| in1 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| in2 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| in3 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| in4 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| in5 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| linear (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| longMessage (defined in expecttest.TestCase) | expecttest.TestCase | static |
| lstm1 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| lstm2 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| main (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| maxDiff (defined in common_utils.TestCase) | common_utils.TestCase | static |
| out (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| pixel_shuffle (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| precision (defined in common_utils.TestCase) | common_utils.TestCase | static |
| projection (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| reflection_pad (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| relu (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| res1 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| res2 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| res3 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| res4 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| res5 (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| rnn (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| 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_dcgan_models(self) (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| test_dcgan_models_cuda(self) (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| test_mnist(self) (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| test_mnist_cuda(self) (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| test_mnist_training_leaks_no_memory_cuda(self) (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| test_neural_style(self) (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| test_neural_style_cuda(self) (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| test_reinforcement_learning(self) (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| test_reinforcement_learning_cuda(self) (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| test_snli(self) (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| test_snli_cuda(self) (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| test_snli_quantized(self) (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| test_super_resolution(self) (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| test_super_resolution_cuda(self) (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| test_time_sequence_prediction(self) (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| test_vae(self) (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| test_vae_cuda(self) (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| test_vae_quantized(self) (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| upsample (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| upsample_layer (defined in test_jit.TestEndToEndHybridFrontendModels) | test_jit.TestEndToEndHybridFrontendModels | |
| wrap_with_cuda_memory_check(self, method) (defined in common_utils.TestCase) | common_utils.TestCase | |