Home
last modified time | relevance | path

Searched refs:SetNumWorkers (Results 1 – 25 of 38) sorted by relevance

12

/third_party/mindspore/tests/ut/cpp/dataset/
Dc_api_audio_r_to_z_test.cc39 ds = ds->SetNumWorkers(4); in TEST_F()
78 ds = ds->SetNumWorkers(4); in TEST_F()
103 ds = ds->SetNumWorkers(4); in TEST_F()
145 ds = ds->SetNumWorkers(4); in TEST_F()
167 ds = ds->SetNumWorkers(4); in TEST_F()
205 ds = ds->SetNumWorkers(4); in TEST_F()
Doptimization_pass_test.cc46 std::shared_ptr<Dataset> map_leaf = ImageFolder("dir")->SetNumWorkers(0); in TEST_F()
47 std::shared_ptr<Dataset> nonmap_leaf = RandomData(44, schema)->SetNumWorkers(0); in TEST_F()
48 std::shared_ptr<Dataset> batch = Zip({map_leaf, nonmap_leaf})->Batch(1)->SetNumWorkers(0); in TEST_F()
49 std::shared_ptr<Dataset> map = batch->Map({std::shared_ptr<TensorTransform>()})->SetNumWorkers(0); in TEST_F()
Dc_api_audio_a_to_q_test.cc43 ds = ds->SetNumWorkers(4); in TEST_F()
81 ds = ds->SetNumWorkers(4); in TEST_F()
102 ds = ds->SetNumWorkers(4); in TEST_F()
170 ds = ds->SetNumWorkers(4); in TEST_F()
238 ds = ds->SetNumWorkers(4); in TEST_F()
306 ds = ds->SetNumWorkers(4); in TEST_F()
374 ds = ds->SetNumWorkers(4); in TEST_F()
442 ds = ds->SetNumWorkers(4); in TEST_F()
480 ds = ds->SetNumWorkers(4); in TEST_F()
611 ds = ds->SetNumWorkers(4); in TEST_F()
[all …]
Dc_api_epoch_ctrl_test.cc38 ds = ds->SetNumWorkers(2); in TEST_F()
85 ds = ds->SetNumWorkers(3); in TEST_F()
133 ds = ds->SetNumWorkers(3); in TEST_F()
Dc_api_transforms_test.cc153 ds = ds->SetNumWorkers(2); in TEST_F()
218 ds = ds->SetNumWorkers(2); in TEST_F()
341 ds = ds->SetNumWorkers(2); in TEST_F()
406 ds = ds->SetNumWorkers(1); in TEST_F()
455 ds = ds->SetNumWorkers(1); in TEST_F()
503 ds = ds->SetNumWorkers(1); in TEST_F()
584 ds = ds->SetNumWorkers(3); in TEST_F()
639 ds = ds->SetNumWorkers(2); in TEST_F()
693 ds = ds->SetNumWorkers(2); in TEST_F()
748 ds = ds->SetNumWorkers(2); in TEST_F()
[all …]
Dc_api_dataset_randomdata_test.cc38 ds = ds->SetNumWorkers(4); in TEST_F()
152 ds = ds->SetNumWorkers(1); in TEST_F()
488 ds = ds->SetNumWorkers(3); in TEST_F()
526 ds = ds->SetNumWorkers(2); in TEST_F()
569 std::shared_ptr<Dataset> ds = RandomData(3)->SetNumWorkers(5); in TEST_F()
Dc_api_repeat_test.cc31 ds = ds->SetNumWorkers(8); in TEST_F()
Dc_api_dataset_iterator_test.cc203 std::shared_ptr<Dataset> ds = RandomData(random_data_num_row, schema)->SetNumWorkers(1); in TEST_F()
233 std::shared_ptr<Dataset> ds = RandomData(3, schema)->SetNumWorkers(1); in TEST_F()
/third_party/mindspore/mindspore/lite/minddata/example/
Dtestresize.cpp51 ds = ds->SetNumWorkers(1); in main()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/pre/
Ddeep_copy_pass.cc51 new_node->SetNumWorkers(node->NumWorkers()); in Visit()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/
Dcifar10_node.cc139 (*ds)->SetNumWorkers(json_obj["num_parallel_workers"]); in from_json()
Dcifar100_node.cc138 (*ds)->SetNumWorkers(json_obj["num_parallel_workers"]); in from_json()
Dmnist_node.cc132 (*ds)->SetNumWorkers(json_obj["num_parallel_workers"]); in from_json()
Dalbum_node.cc167 (void)((*ds)->SetNumWorkers(json_obj["num_parallel_workers"])); in from_json()
Dimage_folder_node.cc158 (*ds)->SetNumWorkers(json_obj["num_parallel_workers"]); in from_json()
Dflickr_node.cc168 (*ds)->SetNumWorkers(json_obj["num_parallel_workers"]); in from_json()
Dtext_file_node.cc172 (void)((*ds)->SetNumWorkers(json_obj["num_parallel_workers"])); in from_json()
Dceleba_node.cc207 (*ds)->SetNumWorkers(json_obj["num_parallel_workers"]); in from_json()
Dmanifest_node.cc188 (*ds)->SetNumWorkers(json_obj["num_parallel_workers"]); in from_json()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/cache/
Dcache_main.cc56 .SetNumWorkers(static_cast<int32_t>(strtol(argv[ArgIndex::kNumWorkers], nullptr, ds::kDecimal))) in StartServer()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/post/
Dauto_worker_pass.cc72 p.first->SetNumWorkers(cur_node_num_worker); in RunOnTree()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/
Dbatch_node.cc182 (*result)->SetNumWorkers(json_obj["num_parallel_workers"]); in from_json()
Dmap_node.cc204 (*result)->SetNumWorkers(json_obj["num_parallel_workers"]); in from_json()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/
Dcoco_op.h99 Builder &SetNumWorkers(int32_t num_workers) { in SetNumWorkers() function
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/
Dbatch_op.h54 Builder &SetNumWorkers(int32_t num_workers) { in SetNumWorkers() function

12