Home
last modified time | relevance | path

Searched refs:size_getter (Results 1 – 25 of 59) sorted by relevance

123

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/
Dbatch_node.cc119 Status BatchNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estima… in GetDatasetSize() argument
127 RETURN_IF_NOT_OK(size_getter->DryRun(shared_from_this(), dataset_size)); in GetDatasetSize()
133 RETURN_IF_NOT_OK(children_[0]->GetDatasetSize(size_getter, estimate, &num_rows)); in GetDatasetSize()
Dskip_node.cc62 Status SkipNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimat… in GetDatasetSize() argument
69 RETURN_IF_NOT_OK(children_[0]->GetDatasetSize(size_getter, estimate, &num_rows)); in GetDatasetSize()
Dtake_node.cc63 Status TakeNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimat… in GetDatasetSize() argument
70 RETURN_IF_NOT_OK(children_[0]->GetDatasetSize(size_getter, estimate, &num_rows)); in GetDatasetSize()
Dzip_node.cc65 Status ZipNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimate, in GetDatasetSize() argument
74 RETURN_IF_NOT_OK(child->GetDatasetSize(size_getter, estimate, &child_dataset_size)); in GetDatasetSize()
Drepeat_node.cc73 Status RepeatNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estim… in GetDatasetSize() argument
80 RETURN_IF_NOT_OK(children_[0]->GetDatasetSize(size_getter, estimate, &num_rows)); in GetDatasetSize()
Dmap_node.cc210 Status MapNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimate, in GetDatasetSize() argument
221 RETURN_IF_NOT_OK(size_getter->DryRun(shared_from_this(), dataset_size)); in GetDatasetSize()
226 return children_.front()->GetDatasetSize(size_getter, estimate, dataset_size); in GetDatasetSize()
Ddataset_node.cc614 Status DatasetNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool esti… in GetDatasetSize() argument
621 RETURN_IF_NOT_OK(size_getter->DryRun(shared_from_this(), dataset_size)); in GetDatasetSize()
626 return children_.front()->GetDatasetSize(size_getter, estimate, dataset_size); in GetDatasetSize()
631 return children_.back()->GetDatasetSize(size_getter, estimate, dataset_size); in GetDatasetSize()
Dconcat_node.cc81 Status ConcatNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estim… in GetDatasetSize() argument
93 RETURN_IF_NOT_OK(children_[idx]->GetDatasetSize(size_getter, false, &child_dataset_size)); in GetDatasetSize()
Dzip_node.h64 Status GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimate,
Dtake_node.h68 Status GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimate,
Dskip_node.h68 Status GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimate,
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/
Dsbu_node.cc87 Status SBUNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimate, in GetDatasetSize() argument
99 RETURN_IF_NOT_OK(size_getter->DryRun(shared_from_this(), &sample_size)); in GetDatasetSize()
Dcityscapes_node.cc104 Status CityscapesNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool e… in GetDatasetSize() argument
117 RETURN_IF_NOT_OK(size_getter->DryRun(shared_from_this(), &sample_size)); in GetDatasetSize()
Dgenerator_node.cc135 Status GeneratorNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool es… in GetDatasetSize() argument
142 RETURN_IF_NOT_OK(size_getter->DryRun(shared_from_this(), dataset_size)); in GetDatasetSize()
Dcifar10_node.cc87 Status Cifar10Node::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool esti… in GetDatasetSize() argument
101 RETURN_IF_NOT_OK(size_getter->DryRun(shared_from_this(), &sample_size)); in GetDatasetSize()
Ddiv2k_node.cc116 Status DIV2KNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estima… in GetDatasetSize() argument
129 RETURN_IF_NOT_OK(size_getter->DryRun(shared_from_this(), &sample_size)); in GetDatasetSize()
Dcifar100_node.cc89 Status Cifar100Node::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool est… in GetDatasetSize() argument
101 RETURN_IF_NOT_OK(size_getter->DryRun(shared_from_this(), &sample_size)); in GetDatasetSize()
Dmnist_node.cc83 Status MnistNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estima… in GetDatasetSize() argument
95 RETURN_IF_NOT_OK(size_getter->DryRun(shared_from_this(), &sample_size)); in GetDatasetSize()
Dalbum_node.cc96 Status AlbumNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estima… in GetDatasetSize() argument
123 RETURN_IF_NOT_OK(size_getter->DryRun(shared_from_this(), &sample_size)); in GetDatasetSize()
Dimage_folder_node.cc94 Status ImageFolderNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool … in GetDatasetSize() argument
106 RETURN_IF_NOT_OK(size_getter->DryRun(shared_from_this(), &sample_size)); in GetDatasetSize()
Dflickr_node.cc116 Status FlickrNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estim… in GetDatasetSize() argument
129 RETURN_IF_NOT_OK(size_getter->DryRun(shared_from_this(), &sample_size)); in GetDatasetSize()
Dsbu_node.h70 Status GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimate,
Dceleba_node.cc93 Status CelebANode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estim… in GetDatasetSize() argument
165 RETURN_IF_NOT_OK(size_getter->DryRun(shared_from_this(), &sample_size)); in GetDatasetSize()
Dmanifest_node.cc122 Status ManifestNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool est… in GetDatasetSize() argument
138 RETURN_IF_NOT_OK(size_getter->DryRun(shared_from_this(), &sample_size)); in GetDatasetSize()
Dcifar100_node.h69 Status GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimate,

123