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

Fixed minor bug

parent c79ae7d4
......@@ -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
};
......
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