From c3fbce61008f326d2d3043acf289e7310ef93034 Mon Sep 17 00:00:00 2001
From: BrilliantYuKaimin <91609464+BrilliantYuKaimin@users.noreply.github.com>
Date: Mon, 18 Apr 2022 14:34:18 +0800
Subject: [PATCH] Update pixel_unshuffle_sig.cc

---
 paddle/phi/ops/compat/pixel_unshuffle_sig.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/paddle/phi/ops/compat/pixel_unshuffle_sig.cc b/paddle/phi/ops/compat/pixel_unshuffle_sig.cc
index e78b676dd629e..ce2939d13a042 100644
--- a/paddle/phi/ops/compat/pixel_unshuffle_sig.cc
+++ b/paddle/phi/ops/compat/pixel_unshuffle_sig.cc
@@ -25,9 +25,9 @@ KernelSignature PixelUnshuffleOpArgumentMapping(
 KernelSignature PixelUnshuffleGradOpArgumentMapping(
     const ArgumentMappingContext& ctx) {
   return KernelSignature("pixel_unshuffle_grad",
-                         {GradVarName("Out")},
+                         {"Out@GRAD"},
                          {"downscale_factor", "data_format"},
-                         {GradVarName("X")});
+                         {"X@GRAD"});
 }
 
 }  // namespace phi