Home
last modified time | relevance | path

Searched defs:DummyDataset (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/test/cpp/jit/
Dtest_lite_trainer.cpp313 struct DummyDataset : torch::data::datasets::Dataset<DummyDataset, int> { struct
314 explicit DummyDataset(size_t size = 100) : size_(size) {} in DummyDataset() function
316 int get(size_t index) override { in get()
320 torch::optional<size_t> size() const override { in size()
324 size_t size_;
/external/pytorch/test/cpp/api/
Ddataloader.cpp29 struct DummyDataset : datasets::Dataset<DummyDataset, int> { struct
30 explicit DummyDataset(size_t size = 100) : size_(size) {} in DummyDataset() argument
44 DummyDataset d; in TEST() argument
/external/pytorch/test/
Dtest_dataloader.py3106 class DummyDataset(torch.utils.data.Dataset): class