diff --git a/src/mindspore2022/mindspore/python/mindspore/_extends/graph_kernel/expanders/fused_mul_add.py b/src/mindspore2022/mindspore/python/mindspore/_extends/graph_kernel/expanders/fused_mul_add.py index 86f3a4d1..3cad5a0a 100644 --- a/src/mindspore2022/mindspore/python/mindspore/_extends/graph_kernel/expanders/fused_mul_add.py +++ b/src/mindspore2022/mindspore/python/mindspore/_extends/graph_kernel/expanders/fused_mul_add.py @@ -20,9 +20,23 @@ class FusedMulAdd(Expander): """FusedMulAdd expander""" def _expand(self, graph_builder): + """ + 执行扩展操作。 + + Args: + graph_builder (GraphBuilder): 图构建器对象。 + + Returns: + Tensor: 执行加法操作后的结果。 + + """ + # 获取输入 input_x, input_y, input_z = self.inputs + # 发射乘法操作 mul_res = graph_builder.emit('Mul', [input_x, input_y]) + # 发射加法操作 result = graph_builder.emit('Add', [mul_res, input_z]) + # 返回结果 return result