摘要: 转载:https://blog.csdn.net/zx_ros/article/details/123526147 自定义算子的步骤: 1.定义算子属性节点 2.编写算子的输入输出类型推导关系函数 3.使用RELAY_REGISTER_OP宏注册算子 4.实现算子的compute函数 5.注册算子的 阅读全文
posted @ 2022-08-06 23:44 牛犁heart 阅读(513) 评论(0) 推荐(0) 编辑
摘要: 在对TVM:编译流程一文中,从ONNX模型中读取模型并转换为relay IR,其中调用_convert_operator函数关于将onnx算子转换成Relay算子,其中如何实现当时直接跳过去了,本节将以卷积算子为例,看下Relay表达式是如何转换为TOPI算子并结合TVM的scheduler在后端上 阅读全文
posted @ 2022-08-06 23:39 牛犁heart 阅读(1076) 评论(0) 推荐(0) 编辑