Searched defs:TreeConsumer (Results 1 – 3 of 3) sorted by relevance
64 explicit IteratorConsumer(int32_t num_epochs = -1) : TreeConsumer(), num_epochs_(num_epochs) {} in TreeConsumer() function104 …: TreeConsumer(), dataset_path_(dataset_path), num_files_(num_files), dataset_type_(dataset_type) … in TreeConsumer() function157 explicit ToDevice(int32_t num_epochs = -1) : TreeConsumer(), num_epochs_(num_epochs) {} in TreeConsumer() function
37 TreeConsumer::TreeConsumer() { tree_adapter_ = std::make_unique<TreeAdapter>(); } in TreeConsumer() function in mindspore::dataset::TreeConsumer
25 class TreeConsumer; variable