FXE0008: fx-node-to-onnx
此诊断跟踪从FX节点到ONNX 运算符的转换过程。
将FX节点转换为ONNX节点的过程涉及到六种不同类型的节点。
-
placeholder
: 表示模块输入,并与 ONNX 图的输入相对应。 -
call_module
: 表示对子模块的调用,并映射到一个ONNX节点 -
call_method
: 表示一个方法调用,但目前还未实现。 -
call_function
: 表示一个函数调用,期望的调用目标是Core ATen。ATen 到 ONNX 的映射由 ONNXScript torchlib 实现。此指南展示了如何为 call_function FX 节点编写和注册自定义符号函数。 -
get_attr
: 表示当前模块中的属性访问,对应于 ONNX 图中的初始化器。 -
output
: 表示模块的输出,对应于 ONNX 图的输出。
要详细了解每种节点类型的转换过程,请查阅 FxOnnxInterpreter
中的相关实现细节。