|
钻瓜专利网为您找到相关结果 53个,建议您 升级VIP下载更多相关专利
- [发明专利]基于自动化搜索的加速卷积Im2col矩阵乘计算方法-CN202311044060.9在审
-
吴都
-
吴都
-
2023-08-18
-
2023-10-27
-
G06F7/523
- 本发明公开了基于自动化搜索的加速卷积Im2col矩阵乘计算方法,包括如下步骤:1)使用Im2col算法将卷积计算转换成矩阵乘法;2)构造TVM用于搜索最优矩阵乘法的模板;3)手工调优的小矩阵乘法库;4)运行时自动代码生成的方法;本发明的有益效果如下:1)本发明使用自动化搜索最优矩阵乘法的方式,从不同的参数组合中搜索到当前硬件架构和卷积形状下最优的参数组合,所以本发明可以针对不同的硬件架构和卷积形状生成性能较优的矩阵乘法代码,从而高效地完成卷积计算。2)本发明使用类似运行时自动代码生成的方法调用经过专门调优的小矩阵乘法库,该方法可以解决长条形矩阵乘法性能较差的问题,得到性能较好的矩阵乘法的方式,从而高效地完成卷积计算。
- 基于自动化搜索加速卷积im2col矩阵计算方法
|