Searched refs:export_modules_as_functions (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | utils.py | 194 export_modules_as_functions: bool | Collection[type[torch.nn.Module]] = False,
517 export_modules_as_functions=export_modules_as_functions,
1356 export_modules_as_functions: bool | Collection[type[torch.nn.Module]],
1407 if isinstance(export_modules_as_functions, bool) and export_modules_as_functions:
1409 elif isinstance(export_modules_as_functions, set) and export_modules_as_functions:
1421 module_typenames = {_find_typename(v) for v in export_modules_as_functions}
1479 export_modules_as_functions: Any = False,
1512 if export_modules_as_functions and opset_version < 15:
1538 model, export_modules_as_functions
|
| H A D | __init__.py | 170 export_modules_as_functions: bool | Collection[type[torch.nn.Module]] = False,
391 export_modules_as_functions=export_modules_as_functions,
|
| H A D | _experimental.py | 27 export_modules_as_functions: Union[bool, Set[Type[torch.nn.Module]]] = False variable in ExportOptions
|
| H A D | verification.py | 589 export_modules_as_functions = export_options.export_modules_as_functions
599 utils._setup_trace_module_map(model, export_modules_as_functions)
|
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_utility_funs.py | 908 export_modules_as_functions={
946 export_modules_as_functions={torch.nn.CELU},
961 export_modules_as_functions=set(),
975 export_modules_as_functions=True,
1011 export_modules_as_functions={NWithOverloads},
1038 export_modules_as_functions=True,
1071 export_modules_as_functions=True,
1135 export_modules_as_functions=True,
|