diff options
| author | Soumith Chintala <soumith@gmail.com> | 2017-06-07 06:33:51 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-06-07 06:33:51 +0300 |
| commit | 21bc88f1b6634cf742993795c8fc26f0f6782d37 (patch) | |
| tree | 59209712efa74993482317514e360751c189ccc3 | |
| parent | fa185dd3b5841a27726d63926b54bec37a95604b (diff) | |
fix outputHeight <-> outputWidth
| -rw-r--r-- | lib/THNN/generic/SpatialDepthWiseConvolution.c | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/lib/THNN/generic/SpatialDepthWiseConvolution.c b/lib/THNN/generic/SpatialDepthWiseConvolution.c index 750bae0..2abb453 100644 --- a/lib/THNN/generic/SpatialDepthWiseConvolution.c +++ b/lib/THNN/generic/SpatialDepthWiseConvolution.c @@ -196,7 +196,7 @@ void THNN_(SpatialDepthWiseConvolution_updateOutput)( THTensor_(free)(bias); THTensor_(free)(_bias); - THTensor_(resize4d)(output, T, nInputPlane * nOutputPlane, outputWidth, outputHeight); + THTensor_(resize4d)(output, T, nInputPlane * nOutputPlane, outputHeight, outputWidth); if (batch == 0) { THTensor_(select)(output, NULL, 0, 0); |