关于非线性运算的疑问

[复制链接] · 发表于 2023-4-17 11:39:16

想问一下，咱们rknntoolkit2的softmax是npu实现的还是要调用cpu实现？两个手册有点矛盾。还有一个问题就是三角函数这种非线性运算未来有用npu实现的可能性吗，如果不行的话怎么在模型中近似代替呢？

只看该作者 · 发表于 2023-4-17 23:56:33

NPU的MAC计算单元加速 maxpooling、relu、conv。
softmax是用通用计算单元算的，不同芯片会有细微差异

只看该作者 · 发表于 2023-4-22 14:41:56

jefferyzhang 发表于 2023-4-17 23:56
NPU的MAC计算单元加速 maxpooling、relu、conv。
softmax是用通用计算单元算的，不同芯片会有细微差异 ...

好的，谢谢