index 23678bf47e434ecd929147c2c0c7b801cd6836c0..6f943a8e92caf15775e0848fc0a2e0ed8a6feed9 100644 (file)
#define CAFFE_VISION_LAYERS_HPP_
#include <leveldb/db.h>
+#include <pthread.h>
#include <vector>
int N_;
};
+template <typename Dtype>
+void* DataLayerPrefetch(void* layer_pointer);
+
template <typename Dtype>
class DataLayer : public Layer<Dtype> {
+ // The function used to perform prefetching.
+ friend void* DataLayerPrefetch<Dtype>(void*);
+
public:
explicit DataLayer(const LayerParameter& param)
: Layer<Dtype>(param) {}
int datum_height_;
int datum_width_;
int datum_size_;
+ pthread_t thread_;
+ shared_ptr<Blob<Dtype> > prefetch_data_;
+ shared_ptr<Blob<Dtype> > prefetch_label_;
};