Searched defs:TestIndexDataset (Results 1 – 1 of 1) sorted by relevance
797 struct TestIndexDataset struct799 explicit TestIndexDataset(size_t size) : data(size) { in TestIndexDataset() function802 std::vector<int> get_batch(TestIndex index) override { in get_batch()809 torch::optional<size_t> size() const override { in size()812 std::vector<int> data;