Fix strides in Conv functional API (#357)

Signed-off-by: Valeriy Fedyunin <valery.fedyunin@abbyy.com>
neoml-lib · Jun 21, 2021 · aefabd3 · aefabd3
1 parent 81257f9
commit aefabd3
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/NeoML/src/Dnn/Layers/3dConvLayer.cpp b/NeoML/src/Dnn/Layers/3dConvLayer.cpp
@@ -216,11 +216,11 @@ CLayerWrapper<C3dConvLayer> Conv3d( int filterCount,
 
 		result->SetFilterHeight( heightParams.FilterSize );
 		result->SetPaddingHeight( heightParams.Padding );
-		result->SetStrideWidth( heightParams.Stride );
+		result->SetStrideHeight( heightParams.Stride );
 
 		result->SetFilterWidth( widthParams.FilterSize );
 		result->SetPaddingWidth( widthParams.Padding );
-		result->SetStrideHeight( widthParams.Stride );
+		result->SetStrideWidth( widthParams.Stride );
 
 		result->SetFilterDepth( depthParams.FilterSize );
 		result->SetPaddingDepth( depthParams.Padding );

diff --git a/NeoML/src/Dnn/Layers/ChannelwiseConvLayer.cpp b/NeoML/src/Dnn/Layers/ChannelwiseConvLayer.cpp
@@ -169,12 +169,12 @@ CLayerWrapper<CChannelwiseConvLayer> ChannelwiseConv( int filterCount,
 
 		result->SetFilterHeight( heightParams.FilterSize );
 		result->SetPaddingHeight( heightParams.Padding );
-		result->SetStrideWidth( heightParams.Stride );
+		result->SetStrideHeight( heightParams.Stride );
 		result->SetDilationHeight( heightParams.Dilation );
 
 		result->SetFilterWidth( widthParams.FilterSize );
 		result->SetPaddingWidth( widthParams.Padding );
-		result->SetStrideHeight( widthParams.Stride );
+		result->SetStrideWidth( widthParams.Stride );
 		result->SetDilationWidth( widthParams.Dilation );
 
 		result->SetZeroFreeTerm( isZeroFreeTerm );

diff --git a/NeoML/src/Dnn/Layers/ConvLayer.cpp b/NeoML/src/Dnn/Layers/ConvLayer.cpp
@@ -160,12 +160,12 @@ CLayerWrapper<CConvLayer> Conv( int filterCount,
 
 		result->SetFilterHeight( heightParams.FilterSize );
 		result->SetPaddingHeight( heightParams.Padding );
-		result->SetStrideWidth( heightParams.Stride );
+		result->SetStrideHeight( heightParams.Stride );
 		result->SetDilationHeight( heightParams.Dilation );
 
 		result->SetFilterWidth( widthParams.FilterSize );
 		result->SetPaddingWidth( widthParams.Padding );
-		result->SetStrideHeight( widthParams.Stride );
+		result->SetStrideWidth( widthParams.Stride );
 		result->SetDilationWidth( widthParams.Dilation );
 
 		result->SetZeroFreeTerm( isZeroFreeTerm );

diff --git a/NeoML/src/Dnn/Layers/TransposedConvLayer.cpp b/NeoML/src/Dnn/Layers/TransposedConvLayer.cpp
@@ -145,12 +145,12 @@ CLayerWrapper<CTransposedConvLayer> TransposedConv( int filterCount,
 
 		result->SetFilterHeight( heightParams.FilterSize );
 		result->SetPaddingHeight( heightParams.Padding );
-		result->SetStrideWidth( heightParams.Stride );
+		result->SetStrideHeight( heightParams.Stride );
 		result->SetDilationHeight( heightParams.Dilation );
 
 		result->SetFilterWidth( widthParams.FilterSize );
 		result->SetPaddingWidth( widthParams.Padding );
-		result->SetStrideHeight( widthParams.Stride );
+		result->SetStrideWidth( widthParams.Stride );
 		result->SetDilationWidth( widthParams.Dilation );
 
 		result->SetZeroFreeTerm( isZeroFreeTerm );