diff options
author | Francisco Massa <fvsmassa@gmail.com> | 2016-01-31 17:32:30 +0300 |
---|---|---|
committer | Francisco Massa <fvsmassa@gmail.com> | 2016-02-18 01:20:06 +0300 |
commit | b339dad44739267bd79a0eae0fb158bfcab5991b (patch) | |
tree | 796bf8399aaf47c0a3eb215e56057b490528809b /SpatialMaxUnpooling.lua | |
parent | 6f2b4380369aa61a29f1ac2e9f2954cf64ae7172 (diff) |
Add THNN conversion for Spatial* modules
Add THNN conversion of SpatialBatchNormalization, SpatialFractionalMaxPooling and SpatialSubSampling
Add THNN convertion of SpatialConvolutionLocal, SpatialFullConvolution and SpatialUpSamplingNearest
THNN conversion of SpatialMaxUnpooling
Remove unfold from generic
Add functional conversion of SpatialCrossMapLRN
Plus fix in the init.c
Fix
Diffstat (limited to 'SpatialMaxUnpooling.lua')
-rw-r--r-- | SpatialMaxUnpooling.lua | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/SpatialMaxUnpooling.lua b/SpatialMaxUnpooling.lua index 219c556..2d37111 100644 --- a/SpatialMaxUnpooling.lua +++ b/SpatialMaxUnpooling.lua @@ -22,13 +22,24 @@ end function SpatialMaxUnpooling:updateOutput(input) self:setParams() - input.nn.SpatialMaxUnpooling_updateOutput(self, input) + input.THNN.SpatialMaxUnpooling_updateOutput( + input:cdata(), + self.output:cdata(), + self.indices:cdata(), + self.owidth, self.oheight + ) return self.output end function SpatialMaxUnpooling:updateGradInput(input, gradOutput) self:setParams() - input.nn.SpatialMaxUnpooling_updateGradInput(self, input, gradOutput) + input.THNN.SpatialMaxUnpooling_updateGradInput( + input:cdata(), + gradOutput:cdata(), + self.gradInput:cdata(), + self.indices:cdata(), + self.owidth, self.oheight + ) return self.gradInput end |