Lines Matching refs:scripted_module

95         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)
408         ).check_not("LoweredWrapper.test_backend").run(self.scripted_module.other.graph)
412             self.scripted_module.other.submodule.graph
546         jit_output = self.scripted_module.forward(*input)
575         self.scripted_module = torch.jit.script(BasicModuleAdd())
583             "backend_with_compiler_demo", self.scripted_module, compile_spec
681         self.scripted_module = torch.jit.script(
685         self.lowered_module = self.scripted_module
687         buffer = io.BytesIO(self.scripted_module._save_to_buffer_for_lite_interpreter())
777         self.scripted_module = torch.jit.script(self.module)
778         buffer = io.BytesIO(self.scripted_module._save_to_buffer_for_lite_interpreter())
799         self.scripted_module = torch.jit.script(BasicModule())
801             self.scripted_module,