diff options
-rw-r--r-- | lib/THCUNN/generic/SpatialDepthWiseConvolution.cu | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/lib/THCUNN/generic/SpatialDepthWiseConvolution.cu b/lib/THCUNN/generic/SpatialDepthWiseConvolution.cu index 2520ff4..5da6570 100644 --- a/lib/THCUNN/generic/SpatialDepthWiseConvolution.cu +++ b/lib/THCUNN/generic/SpatialDepthWiseConvolution.cu @@ -232,7 +232,7 @@ void THNN_(SpatialDepthWiseConvolution_updateOutput)( THCTensor_(free)(state, _bias); // Transpose output - THCTensor_(resize4d)(state, output, batchSize, nInputPlane * nOutputPlane, outputWidth, outputHeight); + THCTensor_(resize4d)(state, output, batchSize, nInputPlane * nOutputPlane, outputHeight, outputWidth); // Make a contiguous copy of output (OPTIONAL) // THCTensor *_output = THCTensor_(newContiguous)(state, output); |