Merge pull request #275 from thuyen/SFCBW

Add backward compatiblity to SpatialFullConvolution

1 files changed, 3 insertions, 0 deletions

diff --git a/SpatialFullConvolution.lua b/SpatialFullConvolution.lua
index 7d766d4..419d23c 100644
--- a/SpatialFullConvolution.lua
+++ b/SpatialFullConvolution.lua
@@ -306,6 +306,7 @@ end
 
 
 function SpatialFullConvolution:updateOutput(input)
+    self:backCompatibility()
     if not self.weightDesc then self:resetWeightDescriptors() end
     self:createIODescriptors(input)
 
@@ -331,6 +332,7 @@ function SpatialFullConvolution:updateOutput(input)
 end
 
 function SpatialFullConvolution:updateGradInput(input, gradOutput)
+    self:backCompatibility()
     if not self.gradInput then return end
     self.gradInput:resizeAs(input)
 
@@ -352,6 +354,7 @@ function SpatialFullConvolution:updateGradInput(input, gradOutput)
 end
 
 function SpatialFullConvolution:accGradParameters(input, gradOutput, scale)
+    self:backCompatibility()
     self.scaleT = self.scaleT or self.weight.new(1)
     -- this line forces this member to always be on CPU (needed for cudnn)
     self.scaleT = torch.type(self.weight) == 'torch.CudaDoubleTensor'