diff options
Diffstat (limited to 'lib/THCUNN/generic/SpatialDilatedConvolution.cu')
-rw-r--r-- | lib/THCUNN/generic/SpatialDilatedConvolution.cu | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/lib/THCUNN/generic/SpatialDilatedConvolution.cu b/lib/THCUNN/generic/SpatialDilatedConvolution.cu index 1ddfb06..6c446ac 100644 --- a/lib/THCUNN/generic/SpatialDilatedConvolution.cu +++ b/lib/THCUNN/generic/SpatialDilatedConvolution.cu @@ -24,6 +24,7 @@ void THNN_(SpatialDilatedConvolution_updateOutput)( THArgCheck(!bias || weight->size[0] == bias->size[0], 4, "nOutputPlane mismatch in weight and bias"); THArgCheck(kW > 0 && kH > 0, 8, "kernel size should be greater than zero"); THArgCheck(dW > 0 && dH > 0, 10, "stride should be greater than zero"); + THArgCheck(dilationW > 0 && dilationH > 0, 14, "dilation should be greater than 0"); // Params: int nInputPlane = weight->size[1]; |