diff options
author | Francisco Massa <fvsmassa@gmail.com> | 2016-01-27 01:10:52 +0300 |
---|---|---|
committer | Francisco Massa <fvsmassa@gmail.com> | 2016-01-31 15:48:47 +0300 |
commit | dd9fabb3f72f698e6ada46aa5c3642737a69c670 (patch) | |
tree | f27acd359e18378af7d98cc3782e89dbcb893ef3 /SpatialConvolutionMM.lua | |
parent | e2ad1f33fe59c6d31a2b3491efb7506b82bdf5b2 (diff) |
Unify C/Cuda signatures for SpatialConvolutionMM and Spatial(AdaptiveMax,Max)Pooling
unfold has it's own file
Diffstat (limited to 'SpatialConvolutionMM.lua')
-rw-r--r-- | SpatialConvolutionMM.lua | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/SpatialConvolutionMM.lua b/SpatialConvolutionMM.lua index 745496a..46b813f 100644 --- a/SpatialConvolutionMM.lua +++ b/SpatialConvolutionMM.lua @@ -76,6 +76,7 @@ function SpatialConvolutionMM:updateOutput(input) self.weight:cdata(), self.bias:cdata(), self.finput:cdata(), + self.fgradInput:cdata(), self.kW, self.kH, self.dW, self.dH, self.padW, self.padH @@ -111,6 +112,10 @@ function SpatialConvolutionMM:accGradParameters(input, gradOutput, scale) self.gradWeight:cdata(), self.gradBias:cdata(), self.finput:cdata(), + self.fgradInput:cdata(), + self.kW, self.kH, + self.dW, self.dH, + self.padW, self.padH, scale ) end |