diff options
author | Cyprien Noel <cyprien.noel@gmail.com> | 2015-05-18 17:45:20 -0700 |
---|---|---|
committer | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2015-08-09 15:13:10 -0700 |
commit | ddcdc9d711e81312caf127e8aa512c3298101297 (patch) | |
tree | 10108ae18daee202c4d36f1df29aecda2ab20ad9 /src/caffe/util | |
parent | 73b3d13b68bedad9d19f70755b0ee4ef376e2a30 (diff) | |
download | caffeonacl-ddcdc9d711e81312caf127e8aa512c3298101297.tar.gz caffeonacl-ddcdc9d711e81312caf127e8aa512c3298101297.tar.bz2 caffeonacl-ddcdc9d711e81312caf127e8aa512c3298101297.zip |
Persistent prefetch thread
Diffstat (limited to 'src/caffe/util')
-rw-r--r-- | src/caffe/util/blocking_queue.cpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/caffe/util/blocking_queue.cpp b/src/caffe/util/blocking_queue.cpp index 73c9564c..6ab6ba06 100644 --- a/src/caffe/util/blocking_queue.cpp +++ b/src/caffe/util/blocking_queue.cpp @@ -1,6 +1,7 @@ #include <boost/thread.hpp> #include <string> +#include "caffe/data_layers.hpp" #include "caffe/util/blocking_queue.hpp" namespace caffe { @@ -83,4 +84,7 @@ size_t BlockingQueue<T>::size() const { return queue_.size(); } +template class BlockingQueue<Batch<float>*>; +template class BlockingQueue<Batch<double>*>; + } // namespace caffe |