Home
last modified time | relevance | path

Searched refs:onnxscript_apis (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/onnx/_internal/exporter/
H A D_compat.py12 from torch.onnx._internal._lazy_import import onnxscript_apis, onnxscript_ir as ir
203 onnx_program.model = onnxscript_apis.convert_version(
206 onnx_program.model = onnxscript_apis.optimize(onnx_program.model)
H A D_registration.py26 from torch.onnx._internal._lazy_import import onnxscript_apis
146 torchlib_ops = onnxscript_apis.get_torchlib_ops()
H A D_onnx_program.py18 from torch.onnx._internal._lazy_import import onnx, onnxscript_apis, onnxscript_ir as ir
172 onnxscript_apis.save_model_with_external_data(self.model, destination)
H A D_core.py25 from torch.onnx._internal._lazy_import import onnxscript_apis
1225 onnxscript_apis.check_model(onnx_program.model)
/aosp_15_r20/external/pytorch/torch/onnx/_internal/
H A D_lazy_import.py33 import onnxscript._framework_apis.torch_2_5 as onnxscript_apis namespace
41 onnxscript_apis = _LazyModule("onnxscript._framework_apis.torch_2_5") variable
H A D_exporter_legacy.py137 import onnxscript._framework_apis.torch_2_5 as onnxscript_apis namespace
139 for meta in onnxscript_apis.get_torchlib_ops():