Searched refs:input_data_addrs (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/ |
| D | task_info.h | 70 …_t binary_size, const std::vector<uint8_t> &meta_data, const std::vector<void *> &input_data_addrs, in TbeTaskInfo() argument
81 input_data_addrs_(input_data_addrs), in TbeTaskInfo()
94 const std::vector<void *> &input_data_addrs() const { return input_data_addrs_; } in input_data_addrs() function
121 … const std::vector<void *> &input_data_addrs, const std::vector<void *> &output_data_addrs, in AicpuTaskInfo() argument
128 input_data_addrs_(input_data_addrs), in AicpuTaskInfo()
135 const std::vector<void *> &input_data_addrs() const { return input_data_addrs_; } in input_data_addrs() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ascend/ |
| D | akg_ascend_kernel_mod.cc | 108 std::vector<void *> input_data_addrs; in GenTask() local
113 (void)std::transform(std::begin(inputs), std::end(inputs), std::back_inserter(input_data_addrs), in GenTask()
134 input_data_addrs, output_data_addrs, workspace_addrs, NeedDump()); in GenTask()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/task/ |
| D | tbe_task.cc | 68 tensor_device_addrs.insert(tensor_device_addrs.end(), task_info_->input_data_addrs().begin(), in Distribute()
69 task_info_->input_data_addrs().end()); in Distribute()
|
| D | aicpu_task.cc | 53 …io_addrs.insert(io_addrs.end(), task_info_->input_data_addrs().begin(), task_info_->input_data_add… in Distribute()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_mod.cc | 83 std::vector<void *> input_data_addrs; in GenTask() local
88 (void)std::transform(std::begin(inputs), std::end(inputs), std::back_inserter(input_data_addrs), in GenTask()
108 …name_, stream_id, stub_func, block_dim_, args, 0, sm_desc, nullptr, 0, meta_data, input_data_addrs, in GenTask()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/ |
| D | aicpu_kernel_mod.cc | 182 std::vector<void *> input_data_addrs; in GenTask() local
183 (void)std::transform(std::begin(inputs), std::end(inputs), std::back_inserter(input_data_addrs), in GenTask()
199 …unique_name_, stream_id, node_so_, node_name_, node_def_str_, ext_info_, input_data_addrs, output_… in GenTask()
|