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 /SpatialCrossMapLRN.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 'SpatialCrossMapLRN.lua')
-rw-r--r-- | SpatialCrossMapLRN.lua | 22 |
1 files changed, 20 insertions, 2 deletions
diff --git a/SpatialCrossMapLRN.lua b/SpatialCrossMapLRN.lua index 2440cd4..9758c79 100644 --- a/SpatialCrossMapLRN.lua +++ b/SpatialCrossMapLRN.lua @@ -16,7 +16,15 @@ function SpatialCrossMapLRN:updateOutput(input) self.scale = self.scale or input.new() if torch.type(input) == 'torch.CudaTensor' then - input.nn.SpatialCrossMapLRN_updateOutput(self, input) + input.THNN.SpatialCrossMapLRN_updateOutput( + input:cdata(), + self.output:cdata(), + self.scale:cdata(), + self.size, + self.alpha, + self.beta, + self.k + ) else local isBatch = true if input:dim() == 3 then @@ -80,7 +88,17 @@ function SpatialCrossMapLRN:updateGradInput(input, gradOutput) 'Input must be 3D or 4D') if torch.type(input) == 'torch.CudaTensor' then - input.nn.SpatialCrossMapLRN_updateGradInput(self, input, gradOutput) + input.THNN.SpatialCrossMapLRN_updateGradInput( + input:cdata(), + gradOutput:cdata(), + self.gradInput:cdata(), + self.scale:cdata(), + self.output:cdata(), + self.size, + self.alpha, + self.beta, + self.k + ) else local isBatch = true if input:dim() == 3 then |