bolt icon indicating copy to clipboard operation
bolt copied to clipboard

arm cpu dilated conv遇到nchw类型的输入会出错

Open chillingche opened this issue 2 years ago • 0 comments

CONVOLUTION_ALGORITHM_GEMM不支持nchw排布的输入,如果模型第一层是dilated conv,选到CONVOLUTION_ALGORITHM_GEMM会计算错误:https://github.com/huawei-noah/bolt/blob/master/compute/tensor/src/cpu/arm/convolution.cpp#L72

chillingche avatar Jul 12 '22 06:07 chillingche