Searched refs:ImageFolderOp (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/ |
| D | image_folder_op.cc | 27 ImageFolderOp::ImageFolderOp(int32_t num_wkrs, std::string file_dir, int32_t queue_size, bool recur… in ImageFolderOp() function in mindspore::dataset::ImageFolderOp
48 Status ImageFolderOp::PrescanMasterEntry(const std::string &filedir) { in PrescanMasterEntry()
85 Status ImageFolderOp::LoadTensorRow(row_id_type row_id, TensorRow *trow) { in LoadTensorRow()
103 void ImageFolderOp::Print(std::ostream &out, bool show_all) const { in Print()
119 Status ImageFolderOp::GetClassIds(std::map<int32_t, std::vector<int64_t>> *cls_ids) const { in GetClassIds()
149 Status ImageFolderOp::PrescanWorkerEntry(int32_t worker_id) { in PrescanWorkerEntry()
183 Status ImageFolderOp::RecursiveWalkFolder(Path *dir) { in RecursiveWalkFolder()
203 Status ImageFolderOp::StartAsyncWalk() { in StartAsyncWalk()
219 Status ImageFolderOp::LaunchThreadsAndInitOp() { in LaunchThreadsAndInitOp()
231 …tree_->AllTasks()->CreateAsyncTask("walk dir", std::bind(&ImageFolderOp::StartAsyncWalk, this), nu… in LaunchThreadsAndInitOp()
[all …]
|
| D | image_folder_op.h | 54 class ImageFolderOp : public MappableLeafOp {
65 …ImageFolderOp(int32_t num_wkrs, std::string file_dir, int32_t queue_size, bool recursive, bool do_…
70 ~ImageFolderOp() = default;
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/ |
| D | image_folder_node.cc | 78 …auto op = std::make_shared<ImageFolderOp>(num_workers_, dataset_dir_, connector_que_size_, recursi… in Build()
101 RETURN_IF_NOT_OK(ImageFolderOp::CountRowsAndClasses(dataset_dir_, exts_, &num_rows, nullptr, {})); in GetDatasetSize()
|
| /third_party/mindspore/tests/ut/cpp/dataset/ |
| D | image_folder_op_test.cc | 49 std::shared_ptr<ImageFolderOp> ImageFolder(int64_t num_works, int64_t rows, int64_t conns, std::str… in ImageFolder()
62 std::shared_ptr<ImageFolderOp> so =
63 …std::make_shared<ImageFolderOp>(num_works, path, conns, false, decode, ext, map, std::move(schema)…
|
| D | cache_op_test.cc | 478 std::shared_ptr<ImageFolderOp> so = std::make_shared<ImageFolderOp>( in TEST_F()
|
| D | map_op_test.cc | 131 std::shared_ptr<ImageFolderOp> ImageFolder(int64_t num_works, int64_t rows, int64_t conns, std::str…
|