diff options
Diffstat (limited to 'SpatialMaxSampling.lua')
-rw-r--r-- | SpatialMaxSampling.lua | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/SpatialMaxSampling.lua b/SpatialMaxSampling.lua index aa16701..d6dc512 100644 --- a/SpatialMaxSampling.lua +++ b/SpatialMaxSampling.lua @@ -11,12 +11,12 @@ function SpatialMaxSampling:__init(...) self.indices = torch.Tensor() end -function SpatialMaxSampling:forward(input) - input.nn.SpatialMaxSampling_forward(self, input) +function SpatialMaxSampling:updateOutput(input) + input.nn.SpatialMaxSampling_updateOutput(self, input) return self.output end -function SpatialMaxSampling:backward(input, gradOutput) - input.nn.SpatialMaxSampling_backward(self, input, gradOutput) +function SpatialMaxSampling:updateGradInput(input, gradOutput) + input.nn.SpatialMaxSampling_updateGradInput(self, input, gradOutput) return self.gradInput end |