MEC —— 优化内存与速度的卷积计算

1. 背景工作 目前的CNN模型中,全连接层往往在最后一层才会使用。 意思也就是说,网络的主体是由卷积层构成的。 因此,加快卷积层的计算对于整个网络的性能至关重要。 目前,卷积的计算大多采用间接计算的方式,主要有以下三种实现方式: im2col + GEMM。 caffe等很多框架中都使用了这种计算方式,原因是将问题转化为矩阵乘法后可以方便的使用很多矩阵运算库(如MKL、openblas、Eige
相关文章
相关标签/搜索