softmax内置支持

[复制链接] · 发表于 2020-6-23 18:00:27

现在很多注意力模型里有很多softmax操作，rknn是否考虑支持softmax算子

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

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

只看该作者 · 发表于 2020-6-23 18:05:32

本帖最后由 jefferyzhang 于 2020-7-1 09:43 编辑

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