Toybrick

pytorch aten hardwish operator support

rakidis

高级会员

积分
507
楼主
发表于 2021-9-6 19:11:37    查看: 3186|回复: 2 | [复制链接]    打印 | 显示全部楼层
Hello,
When I try to convert a pytorch model with 1.7.0 rknn toolkit, I get the error shown below.

NotImplementedError: The following operators are not implemented: ['aten::hardswish_']

Do you have plan to support this hardswish operator in near future? I saw on twm github that they added support for it.

https://github.com/apache/tvm/pull/7174

Do you have plan to merge it rknn toolkit?

Regards


回复

使用道具 举报

rakidis

高级会员

积分
507
沙发
 楼主| 发表于 2021-9-6 22:13:51 | 显示全部楼层
That's a No I guess then.

Custom Ops are very complicated to implement unfortunately.

Many new pytorch projects use aten operations in their solutions, rockchip must try to support most aten operations
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表