FXE0008: fx-node-to-onnx

此诊断跟踪从FX节点到ONNX 运算符的转换过程。

将FX节点转换为ONNX节点的过程涉及到六种不同类型的节点。

  1. placeholder: 表示模块输入,并与 ONNX 图的输入相对应。

  2. call_module: 表示对子模块的调用,并映射到一个ONNX节点

  3. call_method: 表示一个方法调用,但目前还未实现。

  4. call_function: 表示一个函数调用,期望的调用目标是Core ATen。ATen 到 ONNX 的映射由 ONNXScript torchlib 实现。此指南展示了如何为 call_function FX 节点编写和注册自定义符号函数。

  5. get_attr: 表示当前模块中的属性访问,对应于 ONNX 图中的初始化器。

  6. output: 表示模块的输出,对应于 ONNX 图的输出。

要详细了解每种节点类型的转换过程,请查阅 FxOnnxInterpreter 中的相关实现细节。