Home
last modified time | relevance | path

Searched refs:iterator_ctx_ (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dcache_dataset_ops_test.cc182 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_P()
190 iterator_ctx_.get(), /*parent=*/nullptr, in TEST_P()
197 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_P()
324 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
330 iterator_ctx_.get(), /*parent=*/nullptr, in TEST_P()
344 TF_EXPECT_OK(RestoreIterator(iterator_ctx_.get(), &reader, in TEST_P()
350 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
Dwindow_dataset_op_test.cc300 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
310 iterator_ctx_.get(), /*parent=*/nullptr, in TEST_P()
319 iterator_ctx_.get(), &next_element, &end_of_window_dataset)); in TEST_P()
486 TF_EXPECT_OK(RestoreIterator(iterator_ctx_.get(), &reader, in TEST_P()
494 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
504 iterator_ctx_.get(), /*parent=*/nullptr, in TEST_P()
512 iterator_ctx_.get(), &next_element, &end_of_window_dataset)); in TEST_P()
Dshuffle_dataset_op_test.cc335 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_P()
350 iterator_ctx_.get(), /*parent=*/nullptr, in TEST_P()
356 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_P()
536 TF_EXPECT_OK(RestoreIterator(iterator_ctx_.get(), &reader, in TEST_P()
543 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_P()
Dtensor_dataset_op_test.cc132 iterator_->GetNext(iterator_ctx_.get(), &out_tensors, &end_of_sequence)); in TEST_P()
148 iterator_->GetNext(iterator_ctx_.get(), &out_tensors, &end_of_sequence)); in TEST_P()
246 TF_EXPECT_OK(RestoreIterator(iterator_ctx_.get(), &reader, in TEST_P()
253 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_P()
Drepeat_dataset_op_test.cc150 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
166 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
331 TF_EXPECT_OK(RestoreIterator(iterator_ctx_.get(), &reader, in TEST_P()
337 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
Dsparse_tensor_slice_dataset_op_test.cc176 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
385 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
410 TF_EXPECT_OK(RestoreIterator(iterator_ctx_.get(), &reader, in TEST_P()
Dtensor_slice_dataset_op_test.cc115 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
285 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
320 TF_EXPECT_OK(RestoreIterator(iterator_ctx_.get(), &reader, "Iterator", in TEST_P()
Dpadded_batch_dataset_op_test.cc719 iterator_->GetNext(iterator_ctx_.get(), &out_tensors, &end_of_sequence) in ITERATOR_SAVE_AND_RESTORE_TEST_P()
730 iterator_->GetNext(iterator_ctx_.get(), &out_tensors, &end_of_sequence) in TEST_F()
Dflat_map_dataset_op_test.cc222 iterator_->GetNext(iterator_ctx_.get(), &out_tensors, &end_of_sequence) in ITERATOR_SAVE_AND_RESTORE_TEST_P()
Dshard_dataset_op_test.cc325 iterator_->GetNext(iterator_ctx_.get(), &out_tensors, &end_of_sequence) in ITERATOR_SAVE_AND_RESTORE_TEST_P()
Dfilter_dataset_op_test.cc273 iterator_->GetNext(iterator_ctx_.get(), &out_tensors, &end_of_sequence) in TEST_P()
Dparallel_filter_dataset_op_test.cc417 iterator_->GetNext(iterator_ctx_.get(), &out_tensors, &end_of_sequence) in TEST_P()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dlookup_ops.cc64 iterator_ctx_ = std::make_unique<data::IteratorContext>(std::move(params)); in Init()
70 iterator_ctx_.get(), nullptr, "LookupTable", &iterator_)); in Init()
79 status_ = iterator_->GetNext(iterator_ctx_.get(), &tensors_, &end_of_input); in Next()
103 std::unique_ptr<data::IteratorContext> iterator_ctx_; member in tensorflow::data::experimental::__anonb6ed0e5a0111::DatasetIterator
Dlmdb_dataset_op_test.cc175 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence); in TEST_F()
191 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_F()
198 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence); in TEST_F()
Dio_ops_test.cc170 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_P()
273 TF_EXPECT_OK(RestoreIterator(iterator_ctx_.get(), &reader, in TEST_P()
280 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_P()
Dlist_dataset_op_test.cc209 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
374 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
409 TF_EXPECT_OK(RestoreIterator(iterator_ctx_.get(), &reader, "Iterator", in TEST_P()
Drandom_dataset_op_test.cc134 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_P()
Ddirected_interleave_dataset_op_test.cc452 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence).code(), in TEST_F()
/external/tensorflow/tensorflow/core/data/
Ddataset_test_base.cc583 return CheckIteratorGetNext(iterator_.get(), iterator_ctx_.get(), in CheckIteratorGetNext()
619 IteratorContext* ctx = iterator_ctx_.get(); in CheckIteratorSkip()
777 return CheckIteratorSaveAndRestore(dataset_, iterator_ctx_.get(), in CheckIteratorSaveAndRestore()
791 TF_RETURN_IF_ERROR(CreateIteratorContext(dataset_ctx_.get(), &iterator_ctx_)); in Initialize()
793 dataset_->MakeIterator(iterator_ctx_.get(), /*parent=*/nullptr, in Initialize()
Ddataset_test_base.h835 std::unique_ptr<IteratorContext> iterator_ctx_; variable