Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Dcache_dataset_ops_test.cc177 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_P()
185 iterator_ctx_.get(), /*parent=*/nullptr, in TEST_P()
192 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_P()
319 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
325 iterator_ctx_.get(), /*parent=*/nullptr, in TEST_P()
339 TF_EXPECT_OK(RestoreIterator(iterator_ctx_.get(), &reader, in TEST_P()
345 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
Dwindow_dataset_op_test.cc295 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
305 iterator_ctx_.get(), /*parent=*/nullptr, in TEST_P()
314 iterator_ctx_.get(), &next_element, &end_of_window_dataset)); in TEST_P()
481 TF_EXPECT_OK(RestoreIterator(iterator_ctx_.get(), &reader, in TEST_P()
489 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
499 iterator_ctx_.get(), /*parent=*/nullptr, in TEST_P()
507 iterator_ctx_.get(), &next_element, &end_of_window_dataset)); in TEST_P()
Dshuffle_dataset_op_test.cc330 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_P()
345 iterator_ctx_.get(), /*parent=*/nullptr, in TEST_P()
351 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_P()
531 TF_EXPECT_OK(RestoreIterator(iterator_ctx_.get(), &reader, in TEST_P()
538 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_P()
Drepeat_dataset_op_test.cc145 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
161 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
326 TF_EXPECT_OK(RestoreIterator(iterator_ctx_.get(), &reader, in TEST_P()
332 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
Dtensor_dataset_op_test.cc129 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
243 TF_EXPECT_OK(RestoreIterator(iterator_ctx_.get(), &reader, in TEST_P()
250 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_P()
Dtensor_slice_dataset_op_test.cc111 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
278 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
313 TF_EXPECT_OK(RestoreIterator(iterator_ctx_.get(), &reader, "Iterator", in TEST_P()
Dsparse_tensor_slice_dataset_op_test.cc172 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
381 TF_EXPECT_OK(iterator_->GetNext(iterator_ctx_.get(), &out_tensors, in TEST_P()
406 TF_EXPECT_OK(RestoreIterator(iterator_ctx_.get(), &reader, in TEST_P()
Ddataset_test_base.cc578 return CheckIteratorGetNext(iterator_.get(), iterator_ctx_.get(), in CheckIteratorGetNext()
614 IteratorContext* ctx = iterator_ctx_.get(); in CheckIteratorSkip()
762 return CheckIteratorSaveAndRestore(dataset_, iterator_ctx_.get(), in CheckIteratorSaveAndRestore()
776 TF_RETURN_IF_ERROR(CreateIteratorContext(dataset_ctx_.get(), &iterator_ctx_)); in Initialize()
778 dataset_->MakeIterator(iterator_ctx_.get(), /*parent=*/nullptr, in Initialize()
Dflat_map_dataset_op_test.cc203 iterator_->GetNext(iterator_ctx_.get(), &out_tensors, &end_of_sequence) in ITERATOR_SAVE_AND_RESTORE_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()
Dfilter_dataset_op_test.cc271 iterator_->GetNext(iterator_ctx_.get(), &out_tensors, &end_of_sequence) in TEST_P()
Dshard_dataset_op_test.cc325 iterator_->GetNext(iterator_ctx_.get(), &out_tensors, &end_of_sequence) in ITERATOR_SAVE_AND_RESTORE_TEST_P()
Ddataset_test_base.h782 std::unique_ptr<IteratorContext> iterator_ctx_; variable
/external/tensorflow/tensorflow/core/kernels/data/experimental/
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()
Drandom_dataset_op_test.cc132 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_P()
Ddirected_interleave_dataset_op_test.cc357 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence).code(), in TEST_F()
/external/tensorflow/tensorflow/core/kernels/
Dlookup_util.cc416 iterator_ctx_ = absl::make_unique<data::IteratorContext>(std::move(params)); in Init()
417 TF_RETURN_IF_ERROR(dataset_->MakeIterator(iterator_ctx_.get(), nullptr, in Init()
426 status_ = iterator_->GetNext(iterator_ctx_.get(), &tensors_, &end_of_input); in Next()
450 std::unique_ptr<data::IteratorContext> iterator_ctx_; member in tensorflow::lookup::DatasetIterator