Searched refs:shared_dataset (Results 1 – 2 of 2) sorted by relevance
32 std::shared_ptr<UnderlyingDataset> shared_dataset) in SharedBatchDataset() argument33 : dataset_(std::move(shared_dataset)) {} in SharedBatchDataset()
767 auto shared_dataset = in TEST() local772 shared_dataset, torch::data::DataLoaderOptions().workers(3)); in TEST()781 auto shared_dataset = in TEST() local784 ASSERT_EQ(shared_dataset.size().value(), 100); in TEST()