Searched refs:PullBasedIteratorConsumer (Results 1 – 4 of 4) sorted by relevance
20 PullBasedIteratorConsumer::PullBasedIteratorConsumer() { tree_adapter_lite_ = std::make_unique<Tree… in PullBasedIteratorConsumer() function in mindspore::dataset::PullBasedIteratorConsumer22 Status PullBasedIteratorConsumer::Init(std::shared_ptr<DatasetNode> root) { in Init()27 std::vector<TensorRow> PullBasedIteratorConsumer::GetRows(int64_t num_rows) { in GetRows()39 Status PullBasedIteratorConsumer::GetNextAsVector(std::vector<TensorPtr> *const out) { in GetNextAsVector()53 Status PullBasedIteratorConsumer::GetNextAsMap(std::unordered_map<std::string, TensorPtr> *const ou… in GetNextAsMap()70 Status PullBasedIteratorConsumer::GetNextAsOrderedPair( in GetNextAsOrderedPair()
33 class PullBasedIteratorConsumer {36 PullBasedIteratorConsumer();38 ~PullBasedIteratorConsumer() = default;
39 class PullBasedIteratorConsumer; variable163 std::unique_ptr<PullBasedIteratorConsumer> pull_consumer_;
140 pull_consumer_ = std::make_unique<PullBasedIteratorConsumer>(); in BuildAndLaunchTree()