Home
last modified time | relevance | path

Searched refs:op_connector_size (Results 1 – 25 of 69) sorted by relevance

123

/third_party/mindspore/tests/ut/cpp/dataset/
Drandom_data_op_test.cc69 auto op_connector_size = cfg->op_connector_size(); in TEST_F() local
72 std::make_shared<RandomDataOp>(1, op_connector_size, 25, std::move(testSchema)); in TEST_F()
123 auto op_connector_size = cfg->op_connector_size(); in TEST_F() local
125 …tr<RandomDataOp> myRandomDataOp = std::make_shared<RandomDataOp>(1, op_connector_size, 0, nullptr); in TEST_F()
156 auto op_connector_size = cfg->op_connector_size(); in TEST_F() local
159 std::make_shared<RandomDataOp>(1, op_connector_size, 10, std::move(testSchema)); in TEST_F()
215 auto op_connector_size = cfg->op_connector_size(); in TEST_F() local
218 std::make_shared<RandomDataOp>(1, op_connector_size, 10, std::move(testSchema)); in TEST_F()
287 auto op_connector_size = cfg->op_connector_size(); in TEST_F() local
290 std::make_shared<RandomDataOp>(4, op_connector_size, 10, std::move(testSchema)); in TEST_F()
[all …]
Dshuffle_op_test.cc57 auto op_connector_size = config_manager->op_connector_size(); in TEST_F() local
62 1, 16, 0, files, std::move(schema), op_connector_size, columns_to_load, false, 1, 0, false); in TEST_F()
68 …r<ShuffleOp> my_shuffle_op = std::make_shared<ShuffleOp>(4, shuffle_seed, op_connector_size, true); in TEST_F()
129 int32_t op_connector_size = config_manager->op_connector_size(); in TEST_F() local
134 1, 16, 0, files, std::move(schema), op_connector_size, columns_to_load, false, 1, 0, false); in TEST_F()
139 …std::shared_ptr<ShuffleOp> my_shuffle_op = std::make_shared<ShuffleOp>(4, 100, op_connector_size, … in TEST_F()
199 auto op_connector_size = config_manager->op_connector_size(); in TEST_F() local
203 …1, 16, 0, files, std::make_unique<DataSchema>(), op_connector_size, columns_to_load, false, 1, 0, … in TEST_F()
208 …ShuffleOp> my_shuffle_op = std::make_shared<ShuffleOp>(100, shuffle_seed, op_connector_size, true); in TEST_F()
271 int32_t op_connector_size = config_manager->op_connector_size(); in TEST_F() local
[all …]
Dmap_op_test.cc112 auto op_connector_size = config_manager->op_connector_size(); in CreateTFReaderOp() local
119 1, 2, 0, files, std::move(schema), op_connector_size, columns_to_load, false, 1, 0, false); in CreateTFReaderOp()
154 auto op_connector_size = config_manager->op_connector_size(); in TEST_F() local
158 std::make_shared<MapOp>(in_columns, out_columns, std::move(my_func_list), 1, op_connector_size); in TEST_F()
204 auto op_connector_size = config_manager->op_connector_size(); in TEST_F() local
209 std::make_shared<MapOp>(in_columns, out_columns, std::move(my_func_list), 1, op_connector_size); in TEST_F()
256 auto op_connector_size = config_manager->op_connector_size(); in TEST_F() local
261 std::make_shared<MapOp>(in_columns, out_columns, std::move(my_func_list), 1, op_connector_size); in TEST_F()
344 auto op_connector_size = config_manager->op_connector_size(); in TEST_F() local
349 std::make_shared<MapOp>(in_columns, out_columns, std::move(my_func_list), 1, op_connector_size); in TEST_F()
[all …]
DtfReader_op_test.cc44 int32_t op_connector_size = config_manager->op_connector_size(); in TEST_F() local
53 …red<TFReaderOp>(num_workers, worker_connector_size, 0, files, std::move(schema), op_connector_size, in TEST_F()
103 int32_t op_connector_size = config_manager->op_connector_size(); in TEST_F() local
110 …red<TFReaderOp>(num_workers, worker_connector_size, 0, files, std::move(schema), op_connector_size, in TEST_F()
158 int32_t op_connector_size = config_manager->op_connector_size(); in TEST_F() local
167 …red<TFReaderOp>(num_workers, worker_connector_size, 0, files, std::move(schema), op_connector_size, in TEST_F()
215 int32_t op_connector_size = config_manager->op_connector_size(); in TEST_F() local
224 …red<TFReaderOp>(num_workers, worker_connector_size, 0, files, std::move(schema), op_connector_size, in TEST_F()
272 int32_t op_connector_size = config_manager->op_connector_size(); in TEST_F() local
281 …red<TFReaderOp>(num_workers, worker_connector_size, 0, files, std::move(schema), op_connector_size, in TEST_F()
[all …]
Dclient_config_test.cc51 ASSERT_EQ(my_conf->op_connector_size(), kCfgOpConnectorSize); in TEST_F()
62 ASSERT_EQ(my_conf->op_connector_size(), 4); in TEST_F()
71 ASSERT_EQ(my_conf->op_connector_size(), kCfgOpConnectorSize); in TEST_F()
91 auto op_connector_size = config_manager->op_connector_size(); in TEST_F() local
96 1, 2, 0, files, std::move(schema), op_connector_size, columns_to_load, false, 1, 0, false); in TEST_F()
Dzip_op_test.cc61 auto op_connector_size = config_manager->op_connector_size(); in TEST_F() local
63 1, 16, 0, files1, std::move(schema), op_connector_size, columns_to_load, false, 1, 0, false); in TEST_F()
71 …1, 1, 0, files2, std::make_unique<DataSchema>(), op_connector_size, columns_to_load, false, 1, 0, … in TEST_F()
139 auto op_connector_size = config_manager->op_connector_size(); in TEST_F() local
144 1, 16, 0, files1, std::move(schema1), op_connector_size, columns_to_load, false, 1, 0, false); in TEST_F()
155 1, 1, 0, files2, std::move(schema2), op_connector_size, columns_to_load, false, 1, 0, false); in TEST_F()
Dir_callback_test.cc142 int32_t op_connector_size = config_manager->op_connector_size(); in TEST_F() local
145 std::make_shared<RandomDataOp>(num_workers, op_connector_size, 44, std::move(schema)); in TEST_F()
153 …e_shared<MapOp>(input_columns, output_columns, std::move(op_list), num_workers, op_connector_size); in TEST_F()
197 int32_t op_connector_size = config_manager->op_connector_size(); in TEST_F() local
203 …std::shared_ptr<RandomDataOp> leaf = std::make_shared<RandomDataOp>(4, op_connector_size, 4, std::… in TEST_F()
211 …e_shared<MapOp>(input_columns, output_columns, std::move(op_list), num_workers, op_connector_size); in TEST_F()
269 int32_t op_connector_size = config_manager->op_connector_size(); in TEST_F() local
276 …std::shared_ptr<RandomDataOp> leaf = std::make_shared<RandomDataOp>(4, op_connector_size, 4, std::… in TEST_F()
284 …e_shared<MapOp>(input_columns, output_columns, std::move(op_list), num_workers, op_connector_size); in TEST_F()
Dfilter_op_test.cc33 int32_t op_connector_size = config_manager->op_connector_size(); in Filter() local
38 std::make_shared<FilterOp>(in_col_names, num_workers, op_connector_size, predicate_func); in Filter()
Dcache_op_test.cc257 int32_t op_connector_size = config_manager->op_connector_size(); in TEST_F() local
259 std::make_shared<RandomDataOp>(4, op_connector_size, 50, std::move(test_schema)); in TEST_F()
276 std::make_shared<CacheOp>(5, op_connector_size, myClient, std::move(seq_sampler)); in TEST_F()
373 int32_t op_connector_size = config_manager->op_connector_size(); in TEST_F() local
375 std::make_shared<RandomDataOp>(4, op_connector_size, 10, std::move(test_schema)); in TEST_F()
389 std::make_shared<CacheOp>(4, op_connector_size, myClient, std::move(seq_sampler)); in TEST_F()
460 int32_t op_connector_size = config_manager->op_connector_size(); in TEST_F() local
462 std::make_shared<CacheLookupOp>(4, op_connector_size, myClient, std::move(seq_sampler)); in TEST_F()
464 …std::shared_ptr<CacheMergeOp> myMergeOp = std::make_shared<CacheMergeOp>(4, op_connector_size, 4, … in TEST_F()
Dconcat_op_test.cc53 auto op_connector_size = config_manager->op_connector_size(); in TEST_F() local
62 …ed<TFReaderOp>(num_workers, worker_connector_size, 0, files, std::move(schema1), op_connector_size, in TEST_F()
74 …ed<TFReaderOp>(num_workers, worker_connector_size, 0, files, std::move(schema2), op_connector_size, in TEST_F()
Dtake_op_test.cc48 auto op_connector_size = config_manager->op_connector_size(); in TEST_F() local
53 …num_workers, 16, 0, files, std::make_unique<DataSchema>(), op_connector_size, columns_to_load, fal… in TEST_F()
Dskip_op_test.cc38 int32_t op_connector_size = config_manager->op_connector_size(); in TEST_F() local
46 …red<TFReaderOp>(num_workers, worker_connector_size, 0, files, std::move(schema), op_connector_size, in TEST_F()
Dproject_op_test.cc44 auto op_connector_size = config_manager->op_connector_size(); in TEST_F() local
51 …num_workers, 16, 0, files, std::move(schema), op_connector_size, columns_to_load, false, 1, 0, fal… in TEST_F()
Drename_op_test.cc54 int32_t op_connector_size = config_manager->op_connector_size(); in TEST_F() local
61 …red<TFReaderOp>(num_workers, worker_connector_size, 0, files, std::move(schema), op_connector_size, in TEST_F()
Dexecution_tree_test.cc101 auto op_connector_size = config_manager->op_connector_size(); in TEST_F() local
106 1, 2, 0, files, std::move(schema), op_connector_size, columns_to_load, false, 1, 0, false); in TEST_F()
Dceleba_op_test.cc52 auto op_connector_size = config_manager->op_connector_size(); in Celeba() local
54 …std::shared_ptr<CelebAOp> so = std::make_shared<CelebAOp>(num_workers, dir, op_connector_size, dec… in Celeba()
Dtext_file_op_test.cc46 int32_t op_connector_size = 2; in TEST_F() local
59 op_connector_size, shuffle_files, num_devices, device_id); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/
Dparallel_op.cc27 ParallelOp::ParallelOp(int32_t num_workers, int32_t op_connector_size, std::shared_ptr<SamplerRT> s… in ParallelOp() argument
28 : DatasetOp(op_connector_size, sampler), in ParallelOp()
39 oc_queue_size_ = std::max(1, op_connector_size * worker_limit / num_workers_); in ParallelOp()
Dpipeline_op.cc22 PipelineOp::PipelineOp(int32_t op_connector_size, std::shared_ptr<SamplerRT> sampler) in PipelineOp() argument
23 : DatasetOp(op_connector_size, sampler) {} in PipelineOp()
Dcache_lookup_op.h36 …CacheLookupOp(int32_t num_workers, int32_t op_connector_size, std::shared_ptr<CacheClient> cache_c… in CacheLookupOp() argument
38 … : CacheBase(num_workers, op_connector_size, cache_client, sampler), SamplerRT(*(sampler.get())) {} in CacheLookupOp()
Dbarrier_op.cc25 BarrierOp::BarrierOp(int32_t op_connector_size, const std::string &condition_name, py::function con… in BarrierOp() argument
26 : PipelineOp(op_connector_size), in BarrierOp()
Dcache_base_op.cc46 CacheBase::CacheBase(int32_t num_workers, int32_t op_connector_size, std::shared_ptr<CacheClient> c… in CacheBase() argument
48 : ParallelOp(num_workers, op_connector_size, std::move(sampler)), in CacheBase()
60 io_block_queues_.Init(num_workers, op_connector_size); in CacheBase()
61 prefetch_queues_.Init(num_prefetchers_, op_connector_size); in CacheBase()
Dbatch_op.h76 Builder &SetOpConnectorSize(int32_t op_connector_size) { in SetOpConnectorSize() argument
77 …builder_op_connector_size_ = (op_connector_size == 0 ? builder_op_connector_size_ : op_connector_s… in SetOpConnectorSize()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/
Dusps_op.cc40 …int32_t num_workers, int32_t worker_connector_size, int64_t num_samples, int32_t op_connector_size, in USPSOp() argument
42 …: NonMappableLeafOp(num_workers, worker_connector_size, num_samples, op_connector_size, shuffle_fi… in USPSOp()
91 int32_t op_connector_size = cfg->op_connector_size(); in CountTotalRows() local
100 op_connector_size, shuffle, num_devices, device_id); in CountTotalRows()
/third_party/mindspore/tests/ut/cpp/dataset/common/
Dcommon.cc121 auto op_connector_size = config_manager->op_connector_size(); in TFReader() local
126 …er_connector_size, 0, files, std::make_unique<mindspore::dataset::DataSchema>(), op_connector_size, in TFReader()

123