自定义算子是如何使用的,和转TensorRT模型的使用方式一样么?
收藏回复举报
自定义算子是如何使用的,和转TensorRT模型的使用方式一样么?
发表于2024-03-08 10:07:01
0 查看

昇腾芯片上,在pt转onnx的时候,只需要记住自定义算子的符号,然后在转om模型的时候,将自定义算子的具体实现源码放到某个地方,就能完成om的转换?

本帖最后由 匿名用户2024/03/08 10:29:41 编辑

我要发帖子