Searched defs:nn_module (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/jit/ |
| H A D | _recursive.py | 75 def make_stub_from_method(nn_module, method_name): argument
185 def infer_concrete_type_builder(nn_module, share_types=True): argument
433 def get_or_create_concrete_type(self, nn_module): argument
481 def get_module_concrete_type(nn_module, share_types=True): argument
536 def create_script_module(nn_module, stubs_fn, share_types=True, is_tracing=False): argument
560 def create_script_module_impl(nn_module, concrete_type, stubs_fn): argument
835 def infer_methods_to_compile(nn_module): argument
897 def get_hook_stubs(nn_module): argument
931 def get_property_stubs(nn_module): argument
951 def interface_script(mod_interface, nn_module): argument
[all …]
|
| /aosp_15_r20/external/pytorch/torch/_export/ |
| H A D | non_strict_utils.py | 115 nn_module, argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | jit_metaprogramming_utils.py | 590 def create_script_module(self, nn_module, constructor_args, *args, **kwargs): argument
|
| H A D | jit_utils.py | 627 def checkModule(self, nn_module, args): argument
|