Searched refs:iterator_ctx (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/data/ |
| D | dataset_test_base.cc | 661 std::unique_ptr<IteratorContext> iterator_ctx; in CheckSplitProviderShardedIteration() local
663 CreateIteratorContext(dataset->op_kernel_context(), &iterator_ctx)); in CheckSplitProviderShardedIteration()
664 IteratorContext::Params iterator_params(iterator_ctx.get()); in CheckSplitProviderShardedIteration()
667 iterator_ctx = std::make_unique<IteratorContext>(iterator_params); in CheckSplitProviderShardedIteration()
672 dataset->dataset(), iterator_ctx.get(), params.iterator_prefix(), in CheckSplitProviderShardedIteration()
739 DatasetBase* dataset, IteratorContext* iterator_ctx, in CheckIteratorSaveAndRestore() argument
744 TF_RETURN_IF_ERROR(dataset->MakeIterator(iterator_ctx, /*parent=*/nullptr, in CheckIteratorSaveAndRestore()
757 TF_EXPECT_OK(RestoreIterator(iterator_ctx, &reader, iterator_prefix, in CheckIteratorSaveAndRestore()
763 iterator->GetNext(iterator_ctx, &next, &end_of_sequence)); in CheckIteratorSaveAndRestore()
876 std::unique_ptr<IteratorContext> iterator_ctx; in MakeIterator() local
[all …]
|
| D | dataset_test_base.h | 680 DatasetBase* dataset, IteratorContext* iterator_ctx,
|