Searched refs:CheckIteratorGetNext (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | optimize_dataset_op_test.cc | 90 TF_EXPECT_OK(CheckIteratorGetNext(expected_outputs, /*compare_order=*/true)); in TEST_F()
|
| D | dataset_test_base.cc | 576 Status DatasetOpsTestBase::CheckIteratorGetNext( in CheckIteratorGetNext() function in tensorflow::data::DatasetOpsTestBase
578 return CheckIteratorGetNext(iterator_.get(), iterator_ctx_.get(), in CheckIteratorGetNext()
582 Status DatasetOpsTestBase::CheckIteratorGetNext( in CheckIteratorGetNext() function in tensorflow::data::DatasetOpsTestBase
585 return CheckIteratorGetNext(iterator->iterator(), iterator->ctx(), in CheckIteratorGetNext()
589 Status DatasetOpsTestBase::CheckIteratorGetNext( in CheckIteratorGetNext() function in tensorflow::data::DatasetOpsTestBase
640 TF_RETURN_IF_ERROR(CheckIteratorGetNext(iterator.get(), expected_outputs, in CheckSplitProviderFullIteration()
|
| D | dataset_test_base.h | 572 Status CheckIteratorGetNext(const std::vector<Tensor>& expected_outputs,
576 Status CheckIteratorGetNext(TestIterator* iterator,
581 Status CheckIteratorGetNext(IteratorBase* iterator, IteratorContext* ctx,
797 CheckIteratorGetNext(test_case.expected_outputs, \
|