SpatialBatchNormalization.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.

SpatialBatchNormalization.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/SpatialBatchNormalization.lua

blob: 17a45b01f9eb48c013e7d5617c9f7aa357e0009f (plain)

local SpatialBatchNormalization, parent =
   torch.class('cudnn.SpatialBatchNormalization', 'cudnn.BatchNormalization')

SpatialBatchNormalization.mode = 'CUDNN_BATCHNORM_SPATIAL'
SpatialBatchNormalization.nDim = 4
SpatialBatchNormalization.__version = 2