diff options
author | Soumith Chintala <soumith@gmail.com> | 2016-02-26 07:06:46 +0300 |
---|---|---|
committer | Soumith Chintala <soumith@gmail.com> | 2016-02-26 07:06:46 +0300 |
commit | 6f9543d2fc0a8fd2c15a420ee0f5fc7e0e341ff9 (patch) | |
tree | 42e6fca97455861c76cc23cf90f9e654d573b0ce /THNN.lua | |
parent | 738b77734bd450cfc2802ae4d83deba1e5df2e40 (diff) | |
parent | c327f5b86ad7972f12fd5fe41c7215eea381043d (diff) |
Merge pull request #665 from colesbury/bn
Add VolumetricBatchNormalization
Diffstat (limited to 'THNN.lua')
-rw-r--r-- | THNN.lua | 4 |
1 files changed, 2 insertions, 2 deletions
@@ -474,7 +474,7 @@ TH_API void THNN_(TemporalSubSampling_accGradParameters)( int kW, int dW, real scale); -TH_API void THNN_(SpatialBatchNormalization_updateOutput)( +TH_API void THNN_(BatchNormalization_updateOutput)( THNNState *state, THTensor *input, THTensor *output, @@ -487,7 +487,7 @@ TH_API void THNN_(SpatialBatchNormalization_updateOutput)( bool train, double momentum, double eps); -TH_API void THNN_(SpatialBatchNormalization_backward)( +TH_API void THNN_(BatchNormalization_backward)( THNNState *state, THTensor *input, THTensor *gradOutput, |