Searched refs:StandaloneTaskIterator (Results 1 – 3 of 3) sorted by relevance
71 StandaloneTaskIterator::StandaloneTaskIterator( in StandaloneTaskIterator() function in tensorflow::data::StandaloneTaskIterator76 Status StandaloneTaskIterator::GetNext(std::vector<Tensor>& element, in GetNext()81 int64 StandaloneTaskIterator::Cardinality() const { in Cardinality()
40 class StandaloneTaskIterator : public TaskIterator {45 StandaloneTaskIterator(std::unique_ptr<standalone::Dataset> dataset,
170 auto task_iterator = absl::make_unique<StandaloneTaskIterator>( in EnsureTaskInitialized()