diff options
author | Andreas Köpf <andreas.koepf@xamla.com> | 2016-02-01 01:45:34 +0300 |
---|---|---|
committer | Andreas Köpf <andreas.koepf@xamla.com> | 2016-02-03 02:31:26 +0300 |
commit | 30cb5a431eafbfecc40e5c22e6b959663b1aa437 (patch) | |
tree | 9a9fefd70f2929f1c23f9a58522e6534d71f8d6f /VolumetricMaxUnpooling.lua | |
parent | 4644b8c61fba57fbacf16d4892ef3f1e37ed4bfd (diff) |
Add THNN conversion of Volumetric* modules
Diffstat (limited to 'VolumetricMaxUnpooling.lua')
-rw-r--r-- | VolumetricMaxUnpooling.lua | 19 |
1 files changed, 17 insertions, 2 deletions
diff --git a/VolumetricMaxUnpooling.lua b/VolumetricMaxUnpooling.lua index fa8a3cc..57d0ee0 100644 --- a/VolumetricMaxUnpooling.lua +++ b/VolumetricMaxUnpooling.lua @@ -30,13 +30,28 @@ end function VolumetricMaxUnpooling:updateOutput(input) self:setParams() - input.nn.VolumetricMaxUnpooling_updateOutput(self, input) + input.THNN.VolumetricMaxUnpooling_updateOutput( + input:cdata(), + self.output:cdata(), + self.indices:cdata(), + self.otime, self.owidth, self.oheight, + self.dT, self.dW, self.dH, + self.padT, self.padW, self.padH + ) return self.output end function VolumetricMaxUnpooling:updateGradInput(input, gradOutput) self:setParams() - input.nn.VolumetricMaxUnpooling_updateGradInput(self, input, gradOutput) + input.THNN.VolumetricMaxUnpooling_updateGradInput( + input:cdata(), + gradOutput:cdata(), + self.gradInput:cdata(), + self.indices:cdata(), + self.otime, self.owidth, self.oheight, + self.dT, self.dW, self.dH, + self.padT, self.padW, self.padH + ) return self.gradInput end |