Toybrick

标题: softmax内置支持 [打印本页]

作者: iamher0 时间: 2020-6-23 18:00
标题: softmax内置支持
现在很多注意力模型里有很多softmax操作，rknn是否考虑支持softmax算子

如果跑不到npu上，封装一个opencl或neon算子也可以啊

rknn也不能只考虑npu支持的算子，soc里还有CPU和gpu,CPU很多都支持simd

作者: jefferyzhang 时间: 2020-6-23 18:05
本帖最后由 jefferyzhang 于 2020-7-1 09:43 编辑

请把不支持的转换log发出来。这些传统激活函数rknn都有支持，甚至yolov4的mish激活函数都也支持了已经。你报的错应该不是这个op不支持的，看下具体log再说。

欢迎光临 Toybrick (https://t.rock-chips.com/)