explicit Generator(unsigned int seed) : rng_(new caffe::rng_t(seed)) {}
explicit Generator(const caffe::rng_t& other) :
rng_(new caffe::rng_t(other)) {}
+ const caffe::rng_t& rng() const { return *rng_; }
+ private:
shared_ptr<caffe::rng_t> rng_;
};
}
const void* Caffe::RNG::generator() const {
- return static_cast<const void*>(generator_->rng_.get());
+ return static_cast<const void*>(&generator_->rng());
}
void Caffe::RNG::set_generator(const void* other_rng) {