Searched refs:fusion_op (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel_fusion.cc | 76 nlohmann::json fusion_op; in KernelFusion() local
77 …ild::GenFusionScopeJson(fusion_scope_iter.input_nodes, fusion_scope_iter.compute_nodes, &fusion_op, in KernelFusion()
82 size_t hash_id = GenFusionJsonHash(fusion_op); in KernelFusion()
85 fusion_op["fusion_op_name"] = json_name; 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()
117 fusion_json["fusion_op"] = fusion_op; in KernelFusion()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/ |
| D | ub_pattern_fusion.cc | 71 auto fusion_op = std::make_shared<Primitive>(fusion_op_name); in CreateFusionOp() local
72 MS_EXCEPTION_IF_NULL(fusion_op); in CreateFusionOp()
85 fusion_op->set_attr("input_names", input_names_v); in CreateFusionOp()
86 fusion_op->set_attr("output_names", output_names_v); in CreateFusionOp()
92 fusion_op->set_attr(kAttrFracZGroup, MakeValue(fracz_group)); in CreateFusionOp()
97 auto value_node = std::make_shared<ValueNode>(fusion_op); in CreateFusionOp()
|
| /third_party/mindspore/mindspore/_extends/parallel_compile/tbe_compiler/ |
| D | compiler.py | 20 from te_fusion.fusion_util import fusion_op
187 …return fusion_op(json.dumps(fusion_op_arg), reset_op_info=reset_op_info, switch_str=rl_tune_switch,
|
| D | tuner.py | 23 from te_fusion.fusion_util import fusion_op, dump_fusion_json
344 fusion_op(converted_json, reset_op_info=reset_op_info)
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | ascend_kernel_compile.cc | 665 nlohmann::json fusion_op; in AscendFusionOpCompile() local
666 if (!json_creator->GenJson(fusion_scope_iter, &fusion_op)) { in AscendFusionOpCompile()
673 …if (!TbeKernelBuild::GetIOSize(fusion_op[kOpList], fusion_scope_iter.output_nodes, &input_size_lis… in AscendFusionOpCompile()
700 JsonAssemble(job_type, fusion_op, &build_json); in AscendFusionOpCompile()
|