Home
last modified time | relevance | path

Searched refs:create_script_module (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Djit_metaprogramming_utils.py590 def create_script_module(self, nn_module, constructor_args, *args, **kwargs): function
623 create_script_module.last_graph = module.graph # type: ignore[attr-defined]
716 …args, mod = f_args_variable, create_script_module(None, nn_module, constructor_args, *f_args_varia…
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_type_sharing.py560 top1_s = torch.jit._recursive.create_script_module(
565 top2_s = torch.jit._recursive.create_script_module(
H A Dtest_backends.py318 return torch.jit._recursive.create_script_module(
/aosp_15_r20/external/pytorch/torch/jit/
H A D_recursive.py536 def create_script_module(nn_module, stubs_fn, share_types=True, is_tracing=False): function
974 return create_script_module(nn_module, infer_interface_methods_to_compile)
H A D_script.py320 ] = torch.jit._recursive.create_script_module(
1147 return torch.jit._recursive.create_script_module(
H A D_trace.py625 return torch.jit._recursive.create_script_module(
1397 script_module = torch.jit._recursive.create_script_module(
/aosp_15_r20/external/pytorch/test/
H A Dtest_jit.py16089 def create_script_module(*args, **kwargs): function
16119 create_script_module.last_graph = module.graph
16170 check_against_reference(self, create_script_module, create_nn_module,