Searched refs:CreateIteratorContext (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | range_dataset_op_test.cc | 91 TF_ASSERT_OK(CreateIteratorContext(range_context.get(), &iterator_context)); in TEST_P()
276 TF_ASSERT_OK(CreateIteratorContext(range_context.get(), &iterator_context)); in TEST_F()
303 TF_ASSERT_OK(CreateIteratorContext(range_context.get(), &iterator_context)); in TEST_F()
333 TF_ASSERT_OK(CreateIteratorContext(range_context.get(), &iterator_context)); in TEST_F()
377 TF_ASSERT_OK(CreateIteratorContext(range_context.get(), &iterator_context)); in TEST_P()
|
| D | tensor_dataset_op_test.cc | 159 CreateIteratorContext(tensor_dataset_context.get(), &iterator_context)); in TEST_P()
375 CreateIteratorContext(tensor_dataset_context.get(), &iterator_context)); in TEST_P()
407 CreateIteratorContext(tensor_dataset_context.get(), &iterator_context)); in TEST_P()
445 CreateIteratorContext(tensor_dataset_context.get(), &iterator_context)); in TEST_F()
478 CreateIteratorContext(tensor_dataset_context.get(), &iterator_ctx)); in TEST_P()
|
| D | sparse_tensor_slice_dataset_op_test.cc | 154 CreateIteratorContext(dataset_kernel_ctx.get(), &iterator_ctx)); in TEST_F()
354 CreateIteratorContext(dataset_kernel_ctx.get(), &iterator_ctx)); in TEST_F()
393 CreateIteratorContext(dataset_kernel_ctx.get(), &iterator_ctx)); in TEST_F()
428 TF_ASSERT_OK(CreateIteratorContext(dataset_kernel_ctx.get(), &iterator_ctx)); in TEST_F()
463 CreateIteratorContext(dataset_kernel_ctx.get(), &iterator_ctx)); in TEST_F()
|
| D | take_dataset_op_test.cc | 185 CreateIteratorContext(take_dataset_context.get(), &iterator_ctx)); in TEST_P()
393 CreateIteratorContext(take_dataset_context.get(), &iterator_ctx)); in TEST_P()
431 CreateIteratorContext(take_dataset_context.get(), &iterator_ctx)); in TEST_P()
469 CreateIteratorContext(take_dataset_context.get(), &iterator_ctx)); in TEST_P()
511 CreateIteratorContext(take_dataset_context.get(), &iterator_ctx)); in TEST_P()
|
| D | repeat_dataset_op_test.cc | 159 CreateIteratorContext(repeat_dataset_context.get(), &iterator_ctx)); in TEST_P()
386 CreateIteratorContext(repeat_dataset_context.get(), &iterator_ctx)); in TEST_P()
423 CreateIteratorContext(repeat_dataset_context.get(), &iterator_ctx)); in TEST_P()
460 CreateIteratorContext(repeat_dataset_context.get(), &iterator_ctx)); in TEST_P()
503 CreateIteratorContext(repeat_dataset_context.get(), &iterator_ctx)); in TEST_P()
|
| D | map_dataset_op_test.cc | 137 TF_ASSERT_OK(CreateIteratorContext(map_context.get(), &iterator_context)); in TEST_P()
365 TF_ASSERT_OK(CreateIteratorContext(map_context.get(), &iterator_context)); in TEST_F()
398 TF_ASSERT_OK(CreateIteratorContext(map_context.get(), &iterator_context)); in TEST_F()
435 TF_ASSERT_OK(CreateIteratorContext(map_context.get(), &iterator_context)); in TEST_F()
495 TF_ASSERT_OK(CreateIteratorContext(map_context.get(), &iterator_context)); in TEST_P()
|
| D | tensor_slice_dataset_op_test.cc | 165 TF_ASSERT_OK(CreateIteratorContext(tensor_slice_dataset_context.get(), in TEST_F()
422 TF_ASSERT_OK(CreateIteratorContext(tensor_slice_dataset_context.get(), in TEST_F()
471 TF_ASSERT_OK(CreateIteratorContext(tensor_slice_dataset_context.get(), in TEST_F()
511 TF_ASSERT_OK(CreateIteratorContext(tensor_slice_dataset_context.get(), in TEST_F()
555 TF_ASSERT_OK(CreateIteratorContext(tensor_slice_dataset_context.get(), in TEST_F()
|
| D | dataset_test_base.h | 149 Status CreateIteratorContext(
|
| D | dataset_test_base.cc | 109 Status DatasetOpsTestBase::CreateIteratorContext( in CreateIteratorContext() function in tensorflow::data::DatasetOpsTestBase
|
| D | concatenate_dataset_op_test.cc | 162 CreateIteratorContext(dataset_kernel_ctx_.get(), &iterator_ctx_)); in CreateIteratorFromTestCase()
|
| D | zip_dataset_op_test.cc | 147 CreateIteratorContext(dataset_kernel_ctx_.get(), &iterator_ctx_)); in CreateIteratorFromTestCase()
|