-
Notifications
You must be signed in to change notification settings - Fork 25.7k
Closed
Closed
Copy link
Labels
module: onnxRelated to torch.onnxRelated to torch.onnxonnx-triagedtriaged by ONNX teamtriaged by ONNX teamtriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module
Description
In the current implementation of torch.onnx.export(..., dynamo=True), certain TorchScript-based operations are not supported. To ensure a smooth user experience, we should implement an automatic fallback mechanism with a deprecation warning. This fallback should handle cases where only the TorchScript-based exporter can process the export, such as:
- When the input model is a
torch.jit.ScriptModuleortorch.jit.ScriptFunction, as these are specifically designed for TorchScript. - When
custom_opsetsis provided, sincetorch.onnx.export(..., dynamo=True)usescustom_translation_tableinstead. - When
dynamic_axesis provided, because the automatic conversion ofdynamic_axestodynamic_shapesdoes not cover all scenarios.
Metadata
Metadata
Assignees
Labels
module: onnxRelated to torch.onnxRelated to torch.onnxonnx-triagedtriaged by ONNX teamtriaged by ONNX teamtriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module