Searched refs:main_thread_dataset_ (Results 1 – 3 of 3) sorted by relevance
39 main_thread_dataset_(std::move(main_thread_dataset)), in options_()178 return this->main_thread_dataset_->get_batch(std::move(*batch_request)); in next()231 std::unique_ptr<Dataset> main_thread_dataset_; variable
44 [this] { this->worker_thread(*this->main_thread_dataset_); }); in StatefulDataLoader()51 this->main_thread_dataset_->reset(); in reset()
52 this->main_thread_dataset_ = in StatelessDataLoader()