Searched refs:iterator_ctx (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_test_base.cc | 654 std::unique_ptr<IteratorContext> iterator_ctx; in CheckSplitProviderShardedIteration() local
656 CreateIteratorContext(dataset->op_kernel_context(), &iterator_ctx)); in CheckSplitProviderShardedIteration()
657 IteratorContext::Params iterator_params(iterator_ctx.get()); in CheckSplitProviderShardedIteration()
659 iterator_ctx = absl::make_unique<IteratorContext>(iterator_params); in CheckSplitProviderShardedIteration()
664 dataset->dataset(), iterator_ctx.get(), params.iterator_prefix(), in CheckSplitProviderShardedIteration()
724 DatasetBase* dataset, IteratorContext* iterator_ctx, in CheckIteratorSaveAndRestore() argument
729 TF_RETURN_IF_ERROR(dataset->MakeIterator(iterator_ctx, /*parent=*/nullptr, in CheckIteratorSaveAndRestore()
742 TF_EXPECT_OK(RestoreIterator(iterator_ctx, &reader, iterator_prefix, in CheckIteratorSaveAndRestore()
748 iterator->GetNext(iterator_ctx, &next, &end_of_sequence)); in CheckIteratorSaveAndRestore()
861 std::unique_ptr<IteratorContext> iterator_ctx; in MakeIterator() local
[all …]
|
| D | dataset_test_base.h | 631 DatasetBase* dataset, IteratorContext* iterator_ctx,
|