Home
last modified time | relevance | path

Searched refs:scripted_module (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_backends.py95 jit_method = self.scripted_module.__getattr__(function_name)
138 self.scripted_module = torch.jit.script(BasicModule())
140 self.scripted_module,
192 self.scripted_module = torch.jit.script(BasicModule())
195 self.scripted_module,
251 self.scripted_module = torch.jit.script(
335 self.scripted_module = script_without_type_sharing(
373 self.scripted_module.graph
382 ).run(self.scripted_module.sub1.graph)
389 ).run(self.scripted_module.sub2.graph)
[all …]
H A Dtest_save_load_for_op_version.py33 scripted_module = torch.jit.script(m())
35 torch.jit.save(scripted_module, buffer)
40 scripted_module = torch.jit.script(m())
41 buffer = io.BytesIO(scripted_module._save_to_buffer_for_lite_interpreter())
463 scripted_module = torch.jit.load(
467 buffer = io.BytesIO(scripted_module._save_to_buffer_for_lite_interpreter())
546 scripted_module = torch.jit.load(
550 buffer = io.BytesIO(scripted_module._save_to_buffer_for_lite_interpreter())
/aosp_15_r20/external/pytorch/test/jit/xnnpack/
H A Dtest_xnnpack_delegate.py22 scripted_module = torch.jit.script(Module())
26 scripted_module,
37 actual_output = scripted_module(sample_input)
48 scripted_module = torch.jit.script(Module())
64 scripted_module,
83 "xnnpack", scripted_module, mismatch_compile_spec
88 scripted_module,
/aosp_15_r20/external/pytorch/test/typing/pass/
H A Ddisabled_jit.py30 scripted_module = jit.script(nn.Linear(2, 2)) variable
31 assert_type(scripted_module, jit.RecursiveScriptModule)
/aosp_15_r20/external/pytorch/docs/source/notes/
H A Dserialization.rst242 >>> scripted_module = torch.jit.script(MyModule())
243 >>> torch.jit.save(scripted_module, 'mymodule.pt')
277 >>> scripted_module = torch.jit.script(ControlFlowModule(), torch.randn(4))
278 >>> torch.jit.save(scripted_module, 'controlflowmodule_scripted.pt')
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/nn/api/
H A Dremote_module_test.py95 scripted_module = torch.jit.script(module)
96 return scripted_module
/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_workflow_ops.py665 scripted_module = torch.jit.script(fq_module)
670 scripted_module(X)
671 self.assertEqual(fq_module.calculate_qparams(), scripted_module.calculate_qparams())
674 torch.jit.save(scripted_module, buf)
/aosp_15_r20/external/pytorch/test/
H A Dtest_jit_autocast.py475 scripted_module = torch.jit.script(TestModule(2, 3)).cuda()
477 result = scripted_module(input)