diff options
Diffstat (limited to 'src/cpu/operators/CpuConv2d.cpp')
-rw-r--r-- | src/cpu/operators/CpuConv2d.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/cpu/operators/CpuConv2d.cpp b/src/cpu/operators/CpuConv2d.cpp index 447b740989..16ac16b3ba 100644 --- a/src/cpu/operators/CpuConv2d.cpp +++ b/src/cpu/operators/CpuConv2d.cpp @@ -215,6 +215,7 @@ ConvolutionMethod CpuConv2d::get_convolution_method(const ITensorInfo *input, co } } #endif // __ARM_FEATURE_FP16_VECTOR_ARITHMETIC + // For 1x1 convolutions run the default GEMM if(weights->dimension(idx_w) == 1 && weights->dimension(idx_h) == 1) { |