Searched defs:TreeConsumer (Results 1 – 6 of 6) sorted by relevance
130 explicit IteratorConsumer(int32_t num_epochs = -1) : TreeConsumer(num_epochs) {} in TreeConsumer() function172 …: TreeConsumer(), dataset_path_(dataset_path), num_files_(num_files), dataset_type_(dataset_type) … in TreeConsumer() function228 explicit ToDevice(int32_t num_epochs = -1) : TreeConsumer(num_epochs) {} in TreeConsumer() function
39 : TreeConsumer(num_epochs), tree_adapter_lite_(std::make_unique<TreeAdapterLite>()) {} in TreeConsumer() function
51 TreeConsumer::TreeConsumer() : TreeConsumer(1) {} in TreeConsumer() function in mindspore::dataset::TreeConsumer53 TreeConsumer::TreeConsumer(int32_t num_epochs) : num_epochs_(num_epochs) { in TreeConsumer() function in mindspore::dataset::TreeConsumer
24 class TreeConsumer; variable
42 friend TreeConsumer; variable
36 class TreeConsumer; variable