| /third_party/mindspore/tests/ut/cpp/dataset/ |
| D | c_api_datasets_test.cc | 240 EXPECT_EQ(ds->GetNumClasses(), 4); in TEST_F()
241 EXPECT_EQ(ds->GetNumClasses(), 4); in TEST_F()
317 EXPECT_EQ(ds->GetNumClasses(), 2); in TEST_F()
326 EXPECT_EQ(ds->GetNumClasses(), -1); in TEST_F()
|
| D | c_api_dataset_cifar_test.cc | 141 int64_t num_classes = ds->GetNumClasses(); in TEST_F()
155 EXPECT_EQ(ds->GetNumClasses(), -1); in TEST_F()
163 EXPECT_EQ(ds->GetNumClasses(), -1); in TEST_F()
213 int64_t num_classes = ds->GetNumClasses(); in TEST_F()
|
| D | c_api_dataset_sbu_test.cc | 151 EXPECT_EQ(ds->GetNumClasses(), -1); in TEST_F()
159 EXPECT_EQ(ds->GetNumClasses(), -1); in TEST_F()
|
| D | c_api_dataset_manifest_test.cc | 121 EXPECT_EQ(ds1->GetNumClasses(), 2); in TEST_F()
126 EXPECT_EQ(ds2->GetNumClasses(), 3); in TEST_F()
|
| D | c_api_dataset_album_test.cc | 124 int64_t num_classes = ds->GetNumClasses(); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | concat_op.cc | 108 Status ConcatOp::GetNumClasses(int64_t *num_classes) { in GetNumClasses() function in mindspore::dataset::ConcatOp
113 RETURN_IF_NOT_OK(child->GetNumClasses(&tmp_num_classes)); in GetNumClasses()
|
| D | dataset_op.cc | 269 Status DatasetOp::GetNumClasses(int64_t *num_classes) { in GetNumClasses() function in mindspore::dataset::DatasetOp
271 return child_[0]->GetNumClasses(num_classes); in GetNumClasses()
276 return child_[child_.size() - 1]->GetNumClasses(num_classes); in GetNumClasses()
|
| D | concat_op.h | 73 Status GetNumClasses(int64_t *num_classes) override;
|
| D | dataset_op.h | 175 virtual Status GetNumClasses(int64_t *num_classes);
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/ |
| D | manifest_op.h | 74 Status GetNumClasses(int64_t *num_classes) override;
|
| D | image_folder_op.h | 109 Status GetNumClasses(int64_t *num_classes) override;
|
| D | manifest_op.cc | 286 Status ManifestOp::GetNumClasses(int64_t *num_classes) { in GetNumClasses() function in mindspore::dataset::ManifestOp
|
| D | image_folder_op.cc | 311 Status ImageFolderOp::GetNumClasses(int64_t *num_classes) { in GetNumClasses() function in mindspore::dataset::ImageFolderOp
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/consumer/ |
| D | bindings.cc | 69 THROW_IF_ERROR(self.GetNumClasses(&num_classes)); in __anona1cf9bfd0602()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/consumers/ |
| D | tree_consumer.h | 201 Status GetNumClasses(int64_t *num_classes);
|
| D | tree_consumer.cc | 632 Status TreeGetters::GetNumClasses(int64_t *num_classes) { in GetNumClasses() function in mindspore::dataset::TreeGetters
637 RETURN_IF_NOT_OK(root->GetNumClasses(num_classes)); in GetNumClasses()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/mindrecord/include/ |
| D | shard_reader.h | 289 int64_t GetNumClasses(const std::string &category_field);
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/liteapi/include/ |
| D | datasets.h | 102 int64_t GetNumClasses();
|
| /third_party/mindspore/mindspore/ccsrc/minddata/mindrecord/io/ |
| D | shard_reader.cc | 888 int64_t ShardReader::GetNumClasses(const std::string &category_field) { in GetNumClasses() function in mindspore::mindrecord::ShardReader
947 auto num_classes = GetNumClasses(category_field); in CountTotalRows()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/ |
| D | datasets.cc | 341 int64_t Dataset::GetNumClasses() { in GetNumClasses() function in mindspore::dataset::Dataset
352 RETURN_SECOND_IF_ERROR(consumer->GetNumClasses(&num_classes), -1); in GetNumClasses()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/include/dataset/ |
| D | datasets.h | 122 int64_t GetNumClasses();
|
| /third_party/mindspore/mindspore/dataset/engine/ |
| D | datasets.py | 1742 self._num_classes = runtime_getter[0].GetNumClasses()
|