diff options
Diffstat (limited to 'Pooling.lua')
| -rw-r--r-- | Pooling.lua | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/Pooling.lua b/Pooling.lua index 87d56bf..af28a3c 100644 --- a/Pooling.lua +++ b/Pooling.lua @@ -84,7 +84,7 @@ local zero = torch.FloatTensor({0}); function Pooling:updateOutput(input) if not self.poolDesc then self:resetPoolDescriptors() end self:createIODescriptors(input) - errcheck('cudnnPoolingForward', cudnn.handle[cutorch.getDevice()-1], + errcheck('cudnnPoolingForward', cudnn.handle[cutorch.getDevice()-1], self.poolDesc[0], one:data(), self.iDesc[0], input:data(), @@ -102,7 +102,8 @@ function Pooling:updateGradInput(input, gradOutput) end if not self.poolDesc then self:resetPoolDescriptors() end self:createIODescriptors(input) - errcheck('cudnnPoolingBackward', cudnn.handle[cutorch.getDevice()-1], self.poolDesc[0], + errcheck('cudnnPoolingBackward', + cudnn.handle[cutorch.getDevice()-1], self.poolDesc[0], one:data(), self.oDesc[0], self.output:data(), self.oDesc[0], gradOutput:data(), |