SpatialMaxPooling.lua - github.com/soumith/cudnn.torch.git - Unnamed repository; edit this file 'description' to name the repository.

Welcome to mirror list, hosted at ThFree Co, Russian Federation.

SpatialMaxPooling.lua - github.com/soumith/cudnn.torch.git - Unnamed repository; edit this file 'description' to name the repository.

summary refs log tree commit diff

path: root/SpatialMaxPooling.lua

blob: ed7aad660786aafcb4fb126b2a01df84bcace91a (plain)

local SpatialMaxPooling, parent = torch.class('cudnn.SpatialMaxPooling', 'cudnn._Pooling')

function SpatialMaxPooling:updateOutput(input)
   self.mode = 'CUDNN_POOLING_MAX'
   return parent.updateOutput(self, input)
end

function SpatialMaxPooling:__tostring__()
   return nn.SpatialMaxPooling.__tostring__(self)
end