Toybrick

标题: 关于非线性运算的疑问 [打印本页]

作者: 江湖艺人 时间: 2023-4-17 11:39
标题: 关于非线性运算的疑问
想问一下，咱们rknntoolkit2的softmax是npu实现的还是要调用cpu实现？两个手册有点矛盾。还有一个问题就是三角函数这种非线性运算未来有用npu实现的可能性吗，如果不行的话怎么在模型中近似代替呢？

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

作者: 江湖艺人 时间: 2023-4-22 14:41

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

好的，谢谢

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