Home
last modified time | relevance | path

Searched refs:GetDatasetSize (Results 1 – 25 of 87) sorted by relevance

1234

/third_party/mindspore/tests/ut/cpp/dataset/
Dc_api_dataset_sbu_test.cc126 EXPECT_EQ(ds->GetDatasetSize(), 5); in TEST_F()
137 EXPECT_EQ(ds->GetDatasetSize(), 5); in TEST_F()
148 EXPECT_EQ(ds->GetDatasetSize(), 5); in TEST_F()
154 EXPECT_EQ(ds->GetDatasetSize(), 5); in TEST_F()
160 EXPECT_EQ(ds->GetDatasetSize(), 5); in TEST_F()
Dc_api_dataset_usps_test.cc193 EXPECT_EQ(ds->GetDatasetSize(), 3); in TEST_F()
204 EXPECT_EQ(ds->GetDatasetSize(), 3); in TEST_F()
217 EXPECT_EQ(ds->GetDatasetSize(), 3); in TEST_F()
222 EXPECT_EQ(ds->GetDatasetSize(), 3); in TEST_F()
227 EXPECT_EQ(ds->GetDatasetSize(), 3); in TEST_F()
Dc_api_dataset_cifar_test.cc126 EXPECT_EQ(ds->GetDatasetSize(), 10000); in TEST_F()
137 EXPECT_EQ(ds->GetDatasetSize(), 10000); in TEST_F()
152 EXPECT_EQ(ds->GetDatasetSize(), 10000); in TEST_F()
158 EXPECT_EQ(ds->GetDatasetSize(), 10000); in TEST_F()
164 EXPECT_EQ(ds->GetDatasetSize(), 10000); in TEST_F()
226 EXPECT_EQ(ds->GetDatasetSize(), 10); in TEST_F()
Dc_api_datasets_test.cc152 EXPECT_EQ(ds->GetDatasetSize(), 1); in TEST_F()
239 EXPECT_EQ(ds->GetDatasetSize(), 44); in TEST_F()
242 EXPECT_EQ(ds->GetDatasetSize(), 44); in TEST_F()
243 EXPECT_EQ(ds->GetDatasetSize(), 44); in TEST_F()
281 EXPECT_EQ(ds->GetDatasetSize(), 20); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/
Dskip_node.cc62 Status SkipNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimat… in GetDatasetSize() function in mindspore::dataset::SkipNode
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() function in mindspore::dataset::TakeNode
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() function in mindspore::dataset::ZipNode
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() function in mindspore::dataset::RepeatNode
80 RETURN_IF_NOT_OK(children_[0]->GetDatasetSize(size_getter, estimate, &num_rows)); in GetDatasetSize()
Dbatch_node.cc119 Status BatchNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estima… in GetDatasetSize() function in mindspore::dataset::BatchNode
133 RETURN_IF_NOT_OK(children_[0]->GetDatasetSize(size_getter, estimate, &num_rows)); in GetDatasetSize()
Dconcat_node.cc81 Status ConcatNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estim… in GetDatasetSize() function in mindspore::dataset::ConcatNode
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,
Dmap_node.cc210 Status MapNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimate, in GetDatasetSize() function in mindspore::dataset::MapNode
226 return children_.front()->GetDatasetSize(size_getter, estimate, dataset_size); in GetDatasetSize()
Dconcat_node.h64 Status GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimate,
/third_party/mindspore/mindspore/lite/examples/unified_api/src/
Dnet_runner.cc192 std::cout << "DatasetSize is " << train_ds_->GetDatasetSize() << std::endl; in InitDB()
194 if (train_ds_->GetDatasetSize() == 0) { in InitDB()
196 MS_ASSERT(train_ds_->GetDatasetSize() != 0); in InitDB()
/third_party/mindspore/mindspore/lite/examples/train_lenet/src/
Dnet_runner.cc208 std::cout << "DatasetSize is " << train_ds_->GetDatasetSize() << std::endl; in InitDB()
210 if (train_ds_->GetDatasetSize() == 0) { in InitDB()
212 MS_ASSERT(train_ds_->GetDatasetSize() != 0); in InitDB()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/
Dsbu_node.h70 Status GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimate,
Dcifar100_node.h69 Status GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimate,
Dmnist_node.h69 Status GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimate,
Dcifar10_node.h69 Status GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimate,
Dflickr_node.h69 Status GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimate,
Dusps_node.h69 Status GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimate,
Dcityscapes_node.h70 Status GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimate,
Dtext_file_node.h71 Status GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimate,

1234