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

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

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