diff options
| author | Soumith Chintala <soumith@gmail.com> | 2016-02-26 07:06:52 +0300 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2016-02-26 07:06:52 +0300 |
| commit | f3b07db09e5542886d52be28ea577dc53d2d45d4 (patch) | |
| tree | 7041a02dfeb1647e30b467c41a55912cc7f65b34 /THCUNN.lua | |
| parent | 9a297de54ae16146d8f060f492b998c84e1387a2 (diff) | |
| parent | 8cd0609bbac97687969d523d58d86fe06d42ee12 (diff) | |
Merge pull request #224 from colesbury/bn
Add per-activation BatchNormalization implementation.
Diffstat (limited to 'THCUNN.lua')
| -rw-r--r-- | THCUNN.lua | 4 |
1 files changed, 2 insertions, 2 deletions
@@ -396,7 +396,7 @@ TH_API void THNN_CudaTemporalMaxPooling_updateGradInput( THCudaTensor *indices, int kW, int dW); -TH_API void THNN_CudaSpatialBatchNormalization_updateOutput( +TH_API void THNN_CudaBatchNormalization_updateOutput( THCState *state, THCudaTensor *input, THCudaTensor *output, @@ -409,7 +409,7 @@ TH_API void THNN_CudaSpatialBatchNormalization_updateOutput( bool train, double momentum, double eps); -TH_API void THNN_CudaSpatialBatchNormalization_backward( +TH_API void THNN_CudaBatchNormalization_backward( THCState *state, THCudaTensor *input, THCudaTensor *gradOutput, |