Support onnx function with all framework apis
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 条评论