Commit 1717605b authored by Davis King's avatar Davis King

Fixed minor bug

parent c79ae7d4
...@@ -250,9 +250,9 @@ namespace dlib { namespace tt ...@@ -250,9 +250,9 @@ namespace dlib { namespace tt
!*/ !*/
private: private:
#ifdef DLIB_USE_CUDA #ifdef DLIB_USE_CUDA
cuda::batch_normalize_conv_gradient impl; cuda::batch_normalize_gradient impl;
#else #else
cpu::batch_normalize_conv_gradient impl; cpu::batch_normalize_gradient impl;
#endif #endif
}; };
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment