Home
last modified time | relevance | path

Searched refs:iterator_context (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Drange_dataset_op_test.cc90 std::unique_ptr<IteratorContext> iterator_context; in TEST_P() local
91 TF_ASSERT_OK(CreateIteratorContext(range_context.get(), &iterator_context)); in TEST_P()
93 TF_ASSERT_OK(range_dataset->MakeIterator(iterator_context.get(), "Iterator", in TEST_P()
99 TF_EXPECT_OK(iterator->GetNext(iterator_context.get(), &out_tensors, in TEST_P()
275 std::unique_ptr<IteratorContext> iterator_context; in TEST_F() local
276 TF_ASSERT_OK(CreateIteratorContext(range_context.get(), &iterator_context)); in TEST_F()
278 TF_ASSERT_OK(range_dataset->MakeIterator(iterator_context.get(), "Iterator", in TEST_F()
302 std::unique_ptr<IteratorContext> iterator_context; in TEST_F() local
303 TF_ASSERT_OK(CreateIteratorContext(range_context.get(), &iterator_context)); in TEST_F()
305 TF_ASSERT_OK(range_dataset->MakeIterator(iterator_context.get(), "Iterator", in TEST_F()
[all …]
Dmap_dataset_op_test.cc136 std::unique_ptr<IteratorContext> iterator_context; in TEST_P() local
137 TF_ASSERT_OK(CreateIteratorContext(map_context.get(), &iterator_context)); in TEST_P()
140 map_dataset->MakeIterator(iterator_context.get(), "Iterator", &iterator)); in TEST_P()
144 TF_EXPECT_OK(iterator->GetNext(iterator_context.get(), &out_tensors, in TEST_P()
364 std::unique_ptr<IteratorContext> iterator_context; in TEST_F() local
365 TF_ASSERT_OK(CreateIteratorContext(map_context.get(), &iterator_context)); in TEST_F()
368 map_dataset->MakeIterator(iterator_context.get(), "Iterator", &iterator)); in TEST_F()
397 std::unique_ptr<IteratorContext> iterator_context; in TEST_F() local
398 TF_ASSERT_OK(CreateIteratorContext(map_context.get(), &iterator_context)); in TEST_F()
401 map_dataset->MakeIterator(iterator_context.get(), "Iterator", &iterator)); in TEST_F()
[all …]
Dtensor_slice_dataset_op_test.cc164 std::unique_ptr<IteratorContext> iterator_context; in TEST_F() local
166 &iterator_context)); in TEST_F()
168 TF_ASSERT_OK(tensor_slice_dataset->MakeIterator(iterator_context.get(), in TEST_F()
175 TF_EXPECT_OK(iterator->GetNext(iterator_context.get(), &out_tensors, in TEST_F()
421 std::unique_ptr<IteratorContext> iterator_context; in TEST_F() local
423 &iterator_context)); in TEST_F()
425 TF_ASSERT_OK(tensor_slice_dataset->MakeIterator(iterator_context.get(), in TEST_F()
470 std::unique_ptr<IteratorContext> iterator_context; in TEST_F() local
472 &iterator_context)); in TEST_F()
474 TF_ASSERT_OK(tensor_slice_dataset->MakeIterator(iterator_context.get(), in TEST_F()
[all …]
Dtensor_dataset_op_test.cc157 std::unique_ptr<IteratorContext> iterator_context; in TEST_P() local
159 CreateIteratorContext(tensor_dataset_context.get(), &iterator_context)); in TEST_P()
161 TF_ASSERT_OK(tensor_dataset->MakeIterator(iterator_context.get(), "Iterator", in TEST_P()
166 TF_EXPECT_OK(iterator->GetNext(iterator_context.get(), &out_tensors, in TEST_P()
373 std::unique_ptr<IteratorContext> iterator_context; in TEST_P() local
375 CreateIteratorContext(tensor_dataset_context.get(), &iterator_context)); in TEST_P()
377 TF_ASSERT_OK(tensor_dataset->MakeIterator(iterator_context.get(), "Iterator", in TEST_P()
405 std::unique_ptr<IteratorContext> iterator_context; in TEST_P() local
407 CreateIteratorContext(tensor_dataset_context.get(), &iterator_context)); in TEST_P()
409 TF_ASSERT_OK(tensor_dataset->MakeIterator(iterator_context.get(), "Iterator", in TEST_P()
[all …]
Ddataset_test_base.cc111 std::unique_ptr<IteratorContext>* iterator_context) { in CreateIteratorContext() argument
115 *iterator_context = absl::make_unique<IteratorContext>(params); in CreateIteratorContext()
Ddataset_test_base.h151 std::unique_ptr<IteratorContext>* iterator_context);