diff options
| author | Gregory Chanan <gchanan@fb.com> | 2016-12-02 02:42:55 +0300 |
|---|---|---|
| committer | Gregory Chanan <gchanan@fb.com> | 2016-12-02 02:49:53 +0300 |
| commit | 81ca1609fc33a3697ed41c44facbbb43fd7f7855 (patch) | |
| tree | 440cf7a9ce9875961346efc0be1564454b2bef55 | |
| parent | 0d223082f1bfc62e7cc7df58e1a124568cfabbed (diff) | |
Remove unnecessary shape checks in Spatial Pooling modules.
Checks comparing input image sizes to kernel sizes are superseded
by output size checks.
| -rw-r--r-- | lib/THCUNN/generic/SpatialAveragePooling.cu | 5 | ||||
| -rw-r--r-- | lib/THCUNN/generic/SpatialDilatedMaxPooling.cu | 5 |
2 files changed, 0 insertions, 10 deletions
diff --git a/lib/THCUNN/generic/SpatialAveragePooling.cu b/lib/THCUNN/generic/SpatialAveragePooling.cu index 851583d..43add02 100644 --- a/lib/THCUNN/generic/SpatialAveragePooling.cu +++ b/lib/THCUNN/generic/SpatialAveragePooling.cu @@ -27,11 +27,6 @@ static inline void THNN_(SpatialAveragePooling_shapeCheck)( THCUNN_argCheck(state, ndim == 3 || ndim == 4, 2, input, "3D or 4D input tensor expected but got: %s"); - - THArgCheck(input->size[dimw] >= kW - padW && input->size[dimh] >= kH - padH, 2, - "input image (H: %d, W: %d) smaller than kernel " - "size - padding( kH: %d padH: %d kW: %d padW: %d", - input->size[dimh], input->size[dimw], kH, padH, kW, padW); THArgCheck(kW/2 >= padW && kH/2 >= padH, 2, "pad should be smaller than half of kernel size, but got " "padW = %d, padH = %d, kW = %d, kH = %d", diff --git a/lib/THCUNN/generic/SpatialDilatedMaxPooling.cu b/lib/THCUNN/generic/SpatialDilatedMaxPooling.cu index 80dbc3e..786f372 100644 --- a/lib/THCUNN/generic/SpatialDilatedMaxPooling.cu +++ b/lib/THCUNN/generic/SpatialDilatedMaxPooling.cu @@ -33,11 +33,6 @@ static inline void THNN_(SpatialDilatedMaxPooling_shapeCheck)( THCUNN_argCheck(state, ndim == 3 || ndim == 4, 2, input, "3D or 4D input tensor expected but got: %s"); - - THArgCheck(input->size[dimw] >= kW - padW && input->size[dimh] >= kH - padH, 2, - "input image (H: %d, W: %d) smaller than kernel " - "size - padding( kH: %d padH: %d kW: %d padW: %d", - input->size[dimh], input->size[dimw], kH, padH, kW, padW); THArgCheck(kW/2 >= padW && kH/2 >= padH, 2, "pad should be smaller than half of kernel size, but got " "padW = %d, padH = %d, kW = %d, kH = %d", |