cudnn.torch/SpatialMaxPooling.lua at master · soumith/cudnn.torch · GitHub

Sitelet https://github.com/soumith/cudnn.torch/blob/master/SpatialMaxPooling.lua

10 lines (8 loc) · 318 Bytes

local SpatialMaxPooling, parent = torch.class('cudnn.SpatialMaxPooling', 'cudnn._Pooling')
function SpatialMaxPooling:updateOutput(input)
   self.mode = 'CUDNN_POOLING_MAX'
   return parent.updateOutput(self, input)
function SpatialMaxPooling:__tostring__()
   return nn.SpatialMaxPooling.__tostring__(self)