pytorch aten hardwish operator support

[复制链接] · 发表于 2021-9-6 19:11:37

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

只看该作者 · 发表于 2021-9-6 21:59:39

You can try using a custom OP to support it.
Plz see the documentation for more information.

只看该作者 · 发表于 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