Searched refs:fusion_scope_iter (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel_fusion.cc | 74 for (const auto &fusion_scope_iter : fusion_scopes) { in KernelFusion() local
77 …if (!TbeKernelBuild::GenFusionScopeJson(fusion_scope_iter.input_nodes, fusion_scope_iter.compute_n… in KernelFusion()
86 fusion_op["full_name"] = fusion_scope_iter.full_name; in KernelFusion()
90 …if (!TbeKernelBuild::GetIOSize(fusion_op["op_list"], fusion_scope_iter.output_nodes, &input_size_l… in KernelFusion()
99 kernel_mod_ret[fusion_scope_iter.scope_id] = kernel_mod; in KernelFusion()
105 …build_manger->SaveSameFusionOpInfo(fusion_scope_iter.scope_id, json_name, tbe::kProcessorAiCore, i… in KernelFusion()
125 fusion_scope_iter.scope_id); in KernelFusion()
|
| /third_party/mindspore/tests/ut/cpp/stub/kernel/ |
| D | kernel_fusion_stub.cc | 25 for (const auto &fusion_scope_iter : fusion_scopes) { in KernelFusion() local
26 kernel_mod_ret[fusion_scope_iter.scope_id] = std::make_shared<TbeKernelMod>(nullptr); in KernelFusion()
44 for (const auto &fusion_scope_iter : fusion_scopes) { in AscendFusionOpCompile() local
45 kernel_mod_ret[fusion_scope_iter.scope_id] = std::make_shared<TbeKernelMod>(nullptr); in AscendFusionOpCompile()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | ascend_kernel_compile.cc | 664 for (const auto &fusion_scope_iter : fusion_scopes) { in AscendFusionOpCompile() local
666 if (!json_creator->GenJson(fusion_scope_iter, &fusion_op)) { in AscendFusionOpCompile()
667 … MS_LOG(WARNING) << "Generate fusion json failed, fusion info: " << fusion_scope_iter.full_name; in AscendFusionOpCompile()
673 …if (!TbeKernelBuild::GetIOSize(fusion_op[kOpList], fusion_scope_iter.output_nodes, &input_size_lis… in AscendFusionOpCompile()
683 kernel_mode_ret[fusion_scope_iter.scope_id] = kernel_mod; in AscendFusionOpCompile()
691 … build_manager_->SaveSameFusionOpInfo(fusion_scope_iter.scope_id, json_name, tbe::kProcessorAiCore, in AscendFusionOpCompile()
710 fusion_scope_iter.scope_id); in AscendFusionOpCompile()
|