diff options
author | Soumith Chintala <soumith@gmail.com> | 2017-08-26 21:47:55 +0300 |
---|---|---|
committer | Soumith Chintala <soumith@gmail.com> | 2017-08-26 21:47:55 +0300 |
commit | c5972818f19fbc40855137d86d1b175c8dc04a5a (patch) | |
tree | 3678609d52b0bff54e9acb465bc16f1ed75a95e9 | |
parent | 6cfd1dd69bc95813439596724f60513e269d6537 (diff) |
fix VolumetricAveragePooling for changed C interface
-rw-r--r-- | VolumetricAveragePooling.lua | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/VolumetricAveragePooling.lua b/VolumetricAveragePooling.lua index df6d2c4..146304a 100644 --- a/VolumetricAveragePooling.lua +++ b/VolumetricAveragePooling.lua @@ -21,7 +21,9 @@ function VolumetricAveragePooling:updateOutput(input) input:cdata(), self.output:cdata(), self.kT, self.kW, self.kH, - self.dT, self.dW, self.dH + self.dT, self.dW, self.dH, + 0, 0, 0, + false, true ) return self.output end @@ -32,7 +34,9 @@ function VolumetricAveragePooling:updateGradInput(input, gradOutput) gradOutput:cdata(), self.gradInput:cdata(), self.kT, self.kW, self.kH, - self.dT, self.dW, self.dH + self.dT, self.dW, self.dH, + 0, 0, 0, + false, true ) return self.gradInput end |