diff options
author | Gregory Chanan <gchanan@fb.com> | 2016-11-29 22:37:52 +0300 |
---|---|---|
committer | Gregory Chanan <gchanan@fb.com> | 2016-12-06 20:05:59 +0300 |
commit | d94416902bdc793960f06b7c784328fd6af70447 (patch) | |
tree | eba58b076f5147111866bdb48c026f903474c7fb /VolumetricDilatedMaxPooling.lua | |
parent | 4c3fbb4af8ae4c6c48e64db8840cec8878bccfeb (diff) |
Improve shape checks for VolumetricMaxPooling and VolumetricDilatedMaxPooling.
Diffstat (limited to 'VolumetricDilatedMaxPooling.lua')
-rw-r--r-- | VolumetricDilatedMaxPooling.lua | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/VolumetricDilatedMaxPooling.lua b/VolumetricDilatedMaxPooling.lua index f4c8d5b..249b2b5 100644 --- a/VolumetricDilatedMaxPooling.lua +++ b/VolumetricDilatedMaxPooling.lua @@ -41,9 +41,11 @@ function VolumetricDilatedMaxPooling:updateGradInput(input, gradOutput) gradOutput:cdata(), self.gradInput:cdata(), self.indices:cdata(), + self.kT, self.kW, self.kH, self.dT, self.dW, self.dH, self.padT, self.padW, self.padH, - self.dilationT, self.dilationW, self.dilationH + self.dilationT, self.dilationW, self.dilationH, + self.ceil_mode ) return self.gradInput end |