FXE0008:fx-node-to-onnx¶
此诊断跟踪从 FX 节点到 ONNX 算子 的转换过程。
将 FX 节点转换为 ONNX 节点的过程涉及处理六种不同的节点类型
placeholder
:表示模块输入,映射到 ONNX 图输入。call_module
:象征对子模块的调用,映射到 ONNXcall_method
:象征方法调用。尚未实现。call_function
:象征函数调用。核心 ATen 预计为函数调用目标。从 ATen 到 ONNX 的映射由 ONNXScript torchlib 实现。此 指南 展示了如何为 call_function FX 节点编写和注册自定义符号函数。get_attr
:表示当前模块中的属性访问。映射到 ONNX 图初始化程序。output
:表示模块的输出。映射到 ONNX 图输出。
要深入了解每种节点类型是如何转换的,请参阅 FxOnnxInterpreter
中的实现细节。