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 /SpatialSubSampling.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 'SpatialSubSampling.lua')
-rw-r--r-- | SpatialSubSampling.lua | 31 |
1 files changed, 28 insertions, 3 deletions
diff --git a/SpatialSubSampling.lua b/SpatialSubSampling.lua index 19aa2f5..2aab799 100644 --- a/SpatialSubSampling.lua +++ b/SpatialSubSampling.lua @@ -40,15 +40,40 @@ function SpatialSubSampling:reset(stdv) end function SpatialSubSampling:updateOutput(input) - return input.nn.SpatialSubSampling_updateOutput(self, input) + input.THNN.SpatialSubSampling_updateOutput( + input:cdata(), + self.output:cdata(), + self.weight:cdata(), + self.bias:cdata(), + self.kW, self.kH, + self.dW, self.dH + ) + return self.output end function SpatialSubSampling:updateGradInput(input, gradOutput) if self.gradInput then - return input.nn.SpatialSubSampling_updateGradInput(self, input, gradOutput) + input.THNN.SpatialSubSampling_updateGradInput( + input:cdata(), + gradOutput:cdata(), + self.gradInput:cdata(), + self.weight:cdata(), + self.kW, self.kH, + self.dW, self.dH + ) + return self.gradInput end end function SpatialSubSampling:accGradParameters(input, gradOutput, scale) - return input.nn.SpatialSubSampling_accGradParameters(self, input, gradOutput, scale) + scale = scale or 1 + input.THNN.SpatialSubSampling_accGradParameters( + input:cdata(), + gradOutput:cdata(), + self.gradWeight:cdata(), + self.gradBias:cdata(), + self.kW, self.kH, + self.dW, self.dH, + scale + ) end |