diff options
Diffstat (limited to 'lib/THCUNN/generic/SpatialConvolutionMM.cu')
-rw-r--r-- | lib/THCUNN/generic/SpatialConvolutionMM.cu | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/lib/THCUNN/generic/SpatialConvolutionMM.cu b/lib/THCUNN/generic/SpatialConvolutionMM.cu index b4ae8e5..4db0406 100644 --- a/lib/THCUNN/generic/SpatialConvolutionMM.cu +++ b/lib/THCUNN/generic/SpatialConvolutionMM.cu @@ -302,7 +302,7 @@ void THNN_(SpatialConvolutionMM_updateGradInput)( col2im<real, accreal>( THCState_getCurrentStream(state), THCTensor_(data)(state, gradColumns), - nInputPlane, inputHeight, inputWidth, kH, kW, padH, padW, dH, dW, + nInputPlane, inputHeight, inputWidth, outputHeight, outputWidth, kH, kW, padH, padW, dH, dW, 1, 1, THCTensor_(data)(state, gradInput_n) ); } |