| 217 | |
| 218 | template <typename Dtype> |
| 219 | unsigned int WindowDataLayer<Dtype>::PrefetchRand() { |
| 220 | CHECK(prefetch_rng_); |
| 221 | caffe::rng_t* prefetch_rng = |
| 222 | static_cast<caffe::rng_t*>(prefetch_rng_->generator()); |
| 223 | return (*prefetch_rng)(); |
| 224 | } |
| 225 | |
| 226 | // This function is called on prefetch thread |
| 227 | template <typename Dtype> |