/external/tensorflow/tensorflow/core/kernels/data/ |
D | cache_dataset_ops_test.cc | 177 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()
|
D | window_dataset_op_test.cc | 295 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()
|
D | shuffle_dataset_op_test.cc | 330 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()
|
D | repeat_dataset_op_test.cc | 145 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()
|
D | tensor_dataset_op_test.cc | 129 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()
|
D | tensor_slice_dataset_op_test.cc | 111 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()
|
D | sparse_tensor_slice_dataset_op_test.cc | 172 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()
|
D | dataset_test_base.cc | 578 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()
|
D | flat_map_dataset_op_test.cc | 203 iterator_->GetNext(iterator_ctx_.get(), &out_tensors, &end_of_sequence) in ITERATOR_SAVE_AND_RESTORE_TEST_P()
|
D | padded_batch_dataset_op_test.cc | 719 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()
|
D | filter_dataset_op_test.cc | 271 iterator_->GetNext(iterator_ctx_.get(), &out_tensors, &end_of_sequence) in TEST_P()
|
D | shard_dataset_op_test.cc | 325 iterator_->GetNext(iterator_ctx_.get(), &out_tensors, &end_of_sequence) in ITERATOR_SAVE_AND_RESTORE_TEST_P()
|
D | dataset_test_base.h | 782 std::unique_ptr<IteratorContext> iterator_ctx_; variable
|
/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
D | lmdb_dataset_op_test.cc | 175 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()
|
D | random_dataset_op_test.cc | 132 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence)); in TEST_P()
|
D | directed_interleave_dataset_op_test.cc | 357 iterator_->GetNext(iterator_ctx_.get(), &next, &end_of_sequence).code(), in TEST_F()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | lookup_util.cc | 416 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
|