Home
last modified time | relevance | path

Searched refs:DatasetBaseIterator (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_map_iterator.cc33 class ParallelMapIterator : public DatasetBaseIterator {
50 const typename DatasetBaseIterator::BaseParams& base_params, in ParallelMapIterator()
52 : DatasetBaseIterator(base_params), in ParallelMapIterator()
407 const DatasetBaseIterator::BaseParams& params, in NewParallelMapIterator()
Dparallel_map_iterator.h48 const DatasetBaseIterator::BaseParams& params,
/external/tensorflow/tensorflow/core/framework/
Ddataset.h528 friend class DatasetBaseIterator; // for access to `node_` variable
689 class DatasetBaseIterator : public IteratorBase {
699 explicit DatasetBaseIterator(const BaseParams& params) : params_(params) { in DatasetBaseIterator() function
703 ~DatasetBaseIterator() override { params_.dataset->Unref(); } in ~DatasetBaseIterator()
818 class DatasetIterator : public DatasetBaseIterator {
829 : DatasetBaseIterator({params.dataset, params.prefix}), in DatasetIterator()