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: e9a7b89735d859fff83ca84253cdd0913f9b150b (plain)

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

function SpatialMaxPooling:__init(kW, kH, dW, dH, padW, padH)
   parent.__init(self, kW, kH, dW, dH, padW, padH)
   self.mode = 'CUDNN_POOLING_MAX'
end