diff options
Diffstat (limited to 'lib/THCUNN/generic/SpatialDilatedConvolution.cu')
-rw-r--r-- | lib/THCUNN/generic/SpatialDilatedConvolution.cu | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/lib/THCUNN/generic/SpatialDilatedConvolution.cu b/lib/THCUNN/generic/SpatialDilatedConvolution.cu index 01c97c9..ff764f6 100644 --- a/lib/THCUNN/generic/SpatialDilatedConvolution.cu +++ b/lib/THCUNN/generic/SpatialDilatedConvolution.cu @@ -296,7 +296,7 @@ void THNN_(SpatialDilatedConvolution_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, dilationH, dilationW, THCTensor_(data)(state, gradInput_n) ); |