Searched refs:DatasetBaseIterator (Results 1 – 3 of 3) sorted by relevance
81 std::unique_ptr<DatasetBaseIterator> iterator) in State()93 iterator_.reset(static_cast<DatasetBaseIterator*>(it.release())); in DowncastAndSetIterator()112 DatasetBaseIterator* iterator() { return iterator_.get(); } in iterator()121 std::unique_ptr<DatasetBaseIterator> iterator_;
526 DatasetBaseIterator::DatasetBaseIterator(const BaseParams& params) in DatasetBaseIterator() function in tensorflow::data::DatasetBaseIterator532 DatasetBaseIterator::~DatasetBaseIterator() { in ~DatasetBaseIterator()537 string DatasetBaseIterator::BuildTraceMeName() { in BuildTraceMeName()551 Status DatasetBaseIterator::GetNext(IteratorContext* ctx, in GetNext()591 Status DatasetBaseIterator::Skip(IteratorContext* ctx, int num_to_skip, in Skip()626 Status DatasetBaseIterator::SkipInternal(IteratorContext* ctx, int num_to_skip, in SkipInternal()
748 friend class DatasetBaseIterator; // for access to `node_` variable949 class DatasetBaseIterator : public IteratorBase {959 explicit DatasetBaseIterator(const BaseParams& params);961 ~DatasetBaseIterator() override;1103 class DatasetIterator : public DatasetBaseIterator {1114 : DatasetBaseIterator({params.dataset, params.prefix}), in DatasetIterator()