fix:set output name (#51004)

3bba4af7 · feng_shuai · GitHub · 0fd6e2a1 · 3bba4af7
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

paddle/fluid/inference/tensorrt/convert/multihead_matmul_op.cc ...e/fluid/inference/tensorrt/convert/multihead_matmul_op.cc +2 -0

未找到文件。
--- a/paddle/fluid/inference/tensorrt/convert/multihead_matmul_op.cc
+++ b/paddle/fluid/inference/tensorrt/convert/multihead_matmul_op.cc
@@ -674,6 +674,8 @@ class MultiheadMatMulOpConverter : public OpConverter {

          // return
          layer = reshape_after_mha_layer;
+          RreplenishLayerAndOutput(
+              layer, "multihead_matmul", {output_name}, test_mode);
        } else {
          PADDLE_ENFORCE_EQ(
              input->getDimensions().nbDims,