ITADN

Support onnx function with all framework apis

#2792Opentitaiwangms 创建于 2026-02-04
T
titaiwangmscommented
To support `export modules as functions`, we need to extend our current torch framework apis to support functions. https://github.com/microsoft/onnxscript/blob/68e0458e1416ae11f5cd44eb27d07752be5e4960/onnxscript/_framework_apis/torch_2_11.py#L8-L12 The included passes: https://github.com/microsoft/onnxscript/blob/68e0458e1416ae11f5cd44eb27d07752be5e4960/onnxscript/optimizer/_optimizer.py#L45-L73
0 条评论