Update DepthwiseConvolutionLayer.cc

Update prepareF32 condition too.
Samsung · Aug 19, 2024 · 64bdc17 · 64bdc17
1 parent 14261bc
commit 64bdc17
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/runtime/onert/backend/cpu/ops/DepthwiseConvolutionLayer.cc b/runtime/onert/backend/cpu/ops/DepthwiseConvolutionLayer.cc
@@ -30,7 +30,7 @@ namespace ops
 
 void DepthwiseConvolutionLayer::prepareF32()
 {
-  if (_dilationWidth != 1 || _dilationHeight != 1)
+  if (_dilationWidth != 1 || _dilationHeight != 1 || _strideWidth != _strideHeight)
     return;
 
   // DepthwiseConvOp cpu kernel needs additional memory to perform with multi-