Fixed minor bug

This commit is contained in:
Davis King 2015-11-21 09:37:51 -05:00
parent c79ae7d43c
commit 1717605b7b
1 changed files with 2 additions and 2 deletions

View File

@ -250,9 +250,9 @@ namespace dlib { namespace tt
!*/
private:
#ifdef DLIB_USE_CUDA
cuda::batch_normalize_conv_gradient impl;
cuda::batch_normalize_gradient impl;
#else
cpu::batch_normalize_conv_gradient impl;
cpu::batch_normalize_gradient impl;
#endif
};