Searched refs:iterator_ctx_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | concatenate_dataset_op_test.cc | 162 CreateIteratorContext(dataset_kernel_ctx_.get(), &iterator_ctx_)); in CreateIteratorFromTestCase()
164 dataset_->MakeIterator(iterator_ctx_.get(), "Iterator", &iterator_)); in CreateIteratorFromTestCase()
171 std::unique_ptr<IteratorContext> iterator_ctx_; member in tensorflow::data::__anon516b62740111::ConcatenateDatasetOpTestHelper
190 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
324 TF_EXPECT_OK(iterator_->Restore(iterator_ctx_.get(), &reader)); in TEST_P()
327 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
|
| D | zip_dataset_op_test.cc | 147 CreateIteratorContext(dataset_kernel_ctx_.get(), &iterator_ctx_)); in CreateIteratorFromTestCase()
149 dataset_->MakeIterator(iterator_ctx_.get(), "Iterator", &iterator_)); in CreateIteratorFromTestCase()
156 std::unique_ptr<IteratorContext> iterator_ctx_; member in tensorflow::data::__anon6eabbe220111::ZipDatasetOpTestHelper
175 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
323 TF_EXPECT_OK(iterator_->Restore(iterator_ctx_.get(), &reader)); in TEST_P()
326 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
|