专栏算法工具链请问J6M支持自定义onnx算子吗?

请问J6M支持自定义onnx算子吗?

已解决
青芒-阿白2025-05-30
87
4

如题,OE文档里只看到了自定义算子导入,并没有看到开发流程,J5的文档里介绍的自定义CPU算子的流程在J6上适用吗?

算法工具链
征程6
评论2
0/1000
  • Jade-self
    Lv.5
    目前,J6M不太建议使用自定义onnx算子。BPU已经支持了大部分算子,其他不支持的,若不能走项目提需求,可以跑在cpu/DSP上就好。整体建议如下:
    1. 用支持的torch算子拼接成自定义算子
    2. 若1不行,且是模型中间的算子,可以尝试将模型拆段,将自定义算子使用c++运行在CPU上
    3. 若cpu资源不足,J6支持自定义DSP算子
    2025-05-30
    0
    0
  • 费小财
    Lv.5
    你好,可以参考下 下面的文档的
    https://developer.horizon.auto/blog/12599
    2025-05-30
    0
    2
    • 青芒-阿白回复费小财:

      这个文档只是给了一个J6部署onnx模型的例子,并没有介绍任何关于自定义算子的内容,请问J6M支持自定义算子的开发吗?

      2025-05-30
      0
    • 费小财回复青芒-阿白:

      可参考上面的回复

      2025-06-03
      0