- 博客(2)
- 收藏
- 关注
原创 (TVM开发代码学习)熟悉Relay算子的代码
从6个代码块部分熟悉TVM中的Relay算子代码,从添加一个算子的角度去解构TVM中的Relay算子。
2021-12-31 14:53:12 1278
原创 (TVM开发代码学习)给Relay系统添加新算子
本文学习的是官网中的Adding an Operator to Relay — tvm 0.9.dev182+ge718f5a8a documentation (apache.org)添加新的Relay算子需要以下步骤:1、添加一个attribute node,声明在编译时已知的固定参数。2、给算子写一个type relation,整合进Relay类的系统文件里。3、在C++源码中用RELAY_REGISTER_OP宏定义来注册算子的数量、类型等信息。4、定义算子的计算过程。5、给算
2021-12-28 21:29:25 1995
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人