Searched refs:outputs_list (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/ |
| D | ub_pattern_fusion.cc | 63 …teFusionOp(const std::vector<AnfNodePtr> &inputs_list, const std::vector<AnfNodePtr> &outputs_list, in CreateFusionOp() argument
79 for (size_t i = 0; i < outputs_list.size(); i++) { in CreateFusionOp()
109 const std::vector<AnfNodePtr> &outputs_list) { in CreateFusionOpKernelInfo() argument
123 for (const auto &output : outputs_list) { in CreateFusionOpKernelInfo()
187 if (buffer_fusion_info.outputs_list.size() == 1) { // single output in ReplaceOldNode()
189 … kernel_graph->FrontBackendlMapUpdate(buffer_fusion_info.outputs_list[0], buffer_fusion_kernel); in ReplaceOldNode()
191 (void)manager->Replace(buffer_fusion_info.outputs_list[0], buffer_fusion_kernel); in ReplaceOldNode()
192 …laceInputNodeInOtherFusionScope(buffer_fusion_infos, fusion_id, buffer_fusion_info.outputs_list[0], in ReplaceOldNode()
195 for (size_t index = 0; index < buffer_fusion_info.outputs_list.size(); ++index) { in ReplaceOldNode()
198 kernel_graph->FrontBackendlMapUpdate(buffer_fusion_info.outputs_list[index], tuple_item); in ReplaceOldNode()
[all …]
|
| D | fusion_base_pass.h | 54 std::vector<AnfNodePtr> outputs_list; member
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | launch_kernel.cc | 59 …or<kernel::AddressPtr> LaunchKernel::ObtainKernelOutputs(const std::vector<size_t> &outputs_list) { in ObtainKernelOutputs() argument
61 outputs_addr_ = std::vector<uint8_t *>(outputs_list.size(), nullptr); in ObtainKernelOutputs()
62 auto kernel_outputs = ObtainKernelAddress(outputs_list, &outputs_addr_); in ObtainKernelOutputs()
|
| D | launch_kernel.h | 53 std::vector<kernel::AddressPtr> ObtainKernelOutputs(const std::vector<size_t> &outputs_list);
|