Searched refs:make_shared_dataset (Results 1 – 2 of 2) sorted by relevance
109 dataset = datasets::make_shared_dataset<datasets::ChunkDataset< in TEST()768 torch::data::datasets::make_shared_dataset<UncopyableDataset>( in TEST()782 torch::data::datasets::make_shared_dataset<UncopyableDataset>( in TEST()1527 torch::data::datasets::make_shared_dataset<D>(), in TEST()1661 dataset = datasets::make_shared_dataset<datasets::ChunkDataset< in TEST()1717 dataset = datasets::make_shared_dataset<datasets::ChunkDataset< in TEST()1762 dataset = datasets::make_shared_dataset<datasets::ChunkDataset< in TEST()1807 dataset = datasets::make_shared_dataset<datasets::ChunkDataset< in TEST()1843 dataset = datasets::make_shared_dataset<datasets::ChunkDataset< in TEST()1888 dataset = datasets::make_shared_dataset<datasets::ChunkDataset< in TEST()[all …]
78 SharedBatchDataset<UnderlyingDataset> make_shared_dataset(Args&&... args) { in make_shared_dataset() function