diff options
| author | Gregory Chanan <gchanan@fb.com> | 2016-11-03 19:22:27 +0300 |
|---|---|---|
| committer | Gregory Chanan <gchanan@fb.com> | 2016-11-09 00:46:52 +0300 |
| commit | b0444c9da3367e44a0de17d8250742906b47ce46 (patch) | |
| tree | ef9380d5e5ad0af47493ae2be1e3e6d99a7e4b9b /lib/THCUNN/generic/VolumetricDilatedConvolution.cu | |
| parent | 9fd7a885be5db599adcb868494632a3fe0de0205 (diff) | |
ArgCheck that dilation parameters are > 0 and ensure tests
pick dilation parameters > 0.
Diffstat (limited to 'lib/THCUNN/generic/VolumetricDilatedConvolution.cu')
| -rw-r--r-- | lib/THCUNN/generic/VolumetricDilatedConvolution.cu | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/lib/THCUNN/generic/VolumetricDilatedConvolution.cu b/lib/THCUNN/generic/VolumetricDilatedConvolution.cu index b10cc98..22536fe 100644 --- a/lib/THCUNN/generic/VolumetricDilatedConvolution.cu +++ b/lib/THCUNN/generic/VolumetricDilatedConvolution.cu @@ -24,6 +24,7 @@ void THNN_(VolumetricDilatedConvolution_updateOutput)( THArgCheck(!bias || weight->size[0] == bias->size[0], 4, "nOutputPlane mismatch in weight and bias"); THArgCheck(kT > 0 && kW > 0 && kH > 0, 8, "kernel size should be greater than zero"); THArgCheck(dT > 0 && dW > 0 && dH > 0, 10, "stride should be greater than zero"); + THArgCheck(dilationT > 0 && dilationW > 0 && dilationH > 0, 16, "dilation should be greater than 0"); // Params: int nInputPlane = weight->size[1]; |