Searched refs:connector_size (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/tests/ut/cpp/dataset/ |
| D | execution_tree_test.cc | 44 uint32_t connector_size = 8; in TEST_F() local
46 …std::shared_ptr<ShuffleOp> leaf_op1 = std::make_shared<ShuffleOp>(shuffle_size, 0, connector_size,… in TEST_F()
50 …std::shared_ptr<ShuffleOp> leaf_op2 = std::make_shared<ShuffleOp>(shuffle_size, 0, connector_size,… in TEST_F()
54 …red_ptr<ShuffleOp> parent_op = std::make_shared<ShuffleOp>(shuffle_size, 0, connector_size, false); in TEST_F()
66 …std::shared_ptr<DatasetOp> root_op = std::make_shared<ShuffleOp>(shuffle_size, 0, connector_size, … in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/core/ |
| D | config_manager.cc | 134 …onfigManager::set_worker_connector_size(int32_t connector_size) { worker_connector_size_ = connect… in set_worker_connector_size() argument
137 void ConfigManager::set_op_connector_size(int32_t connector_size) { op_connector_size_ = connector_… in set_op_connector_size() argument
|
| D | config_manager.h | 122 void set_worker_connector_size(int32_t connector_size);
126 void set_op_connector_size(int32_t connector_size);
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/ |
| D | device_queue_op.cc | 182 int32_t connector_size = 0; in SendDataToAscend() local
231 connector_capacity, connector_size); in SendDataToAscend()
251 connector_size = ChildOpConnectorSize(); in SendDataToAscend()
280 connector_size = ChildOpConnectorSize(); in SendDataToAscend()
410 int32_t connector_size = 0; in PushDataToGPU() local
471 … profiling_node->Record(CONNECTOR_DEPTH, connector_capacity, send_batch, connector_size, end_time); in PushDataToGPU()
472 connector_size = gpu_item_connector_->size(); in PushDataToGPU()
682 … uint64_t *end_time, int32_t connector_capacity, int32_t connector_size) { in ProfilingRecorder() argument
695 …profiling_node->Record(CONNECTOR_DEPTH, connector_capacity, send_batch + 1, connector_size, *end_t… in ProfilingRecorder()
|
| D | device_queue_op.h | 110 int32_t connector_capacity, int32_t connector_size);
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/perf/ |
| D | CMakeLists.txt | 5 connector_size.cc
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/ |
| D | generator_op.cc | 27 … std::vector<DataType> column_types, int32_t prefetch_size, int32_t connector_size, in GeneratorOp() argument
29 : PipelineOp(connector_size, std::move(sampler)), in GeneratorOp()
|
| D | generator_op.h | 44 std::vector<DataType> column_types, int32_t prefetch_size, int32_t connector_size,
|
| /third_party/mindspore/mindspore/lite/minddata/ |
| D | CMakeLists.txt | 183 ${MINDDATA_DIR}/engine/perf/connector_size.cc
|
| /third_party/mindspore/mindspore/lite/tools/dataset/cropper/ |
| D | dependencies.txt | 1 …connector_size.cc", "mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/album_node.cc", …
|