Searched refs:TestIterator (Results 1 – 2 of 2) sorted by relevance
506 class TestIterator {508 TestIterator(std::unique_ptr<IteratorContext> ctx, in TestIterator() function545 std::unique_ptr<TestIterator>* iterator);549 std::unique_ptr<TestIterator>* iterator);576 Status CheckIteratorGetNext(TestIterator* iterator,
583 TestIterator* iterator, const std::vector<Tensor>& expected_outputs, in CheckIteratorGetNext()637 std::unique_ptr<TestIterator> iterator; in CheckSplitProviderFullIteration()860 std::unique_ptr<TestIterator>* iterator) { in MakeIterator()871 *iterator = std::make_unique<TestIterator>(std::move(iterator_ctx), in MakeIterator()878 std::unique_ptr<TestIterator>* iterator) { in MakeIterator()