| 34 | |
| 35 | template <typename Dtype> |
| 36 | BasePrefetchingDataLayer<Dtype>::BasePrefetchingDataLayer( |
| 37 | const LayerParameter& param) |
| 38 | : BaseDataLayer<Dtype>(param), |
| 39 | prefetch_(param.data_param().prefetch()), |
| 40 | prefetch_free_(), prefetch_full_(), prefetch_current_() { |
| 41 | for (int i = 0; i < prefetch_.size(); ++i) { |
| 42 | prefetch_[i].reset(new Batch<Dtype>()); |
| 43 | prefetch_free_.push(prefetch_[i].get()); |
| 44 | } |
| 45 | } |
| 46 | |
| 47 | template <typename Dtype> |
| 48 | void BasePrefetchingDataLayer<Dtype>::LayerSetUp( |