面向現(xiàn)代GPU的Winograd卷積加速研究
電子學報
頁數(shù): 14 2024-03-12
摘要: 卷積運算是現(xiàn)代卷積神經(jīng)網(wǎng)絡(luò)中必不可少的組成部分,同時也是最耗時的.為了解決卷積算子的性能問題,包括快速傅里葉變換(Fast Fourier Transform,FFT)和Winograd在內(nèi)的快速卷積算法被提出. Winograd卷積可被用于提高小卷積核的推理性能,是目前卷積神經(jīng)網(wǎng)絡(luò)中的主流實現(xiàn)方法 .然而,Winograd卷積在許多高度優(yōu)化的深度神經(jīng)網(wǎng)絡(luò)庫和深度學習編譯器中的... (共14頁)