diff --git a/paddle/phi/ops/compat/channel_shuffle_sig.cc b/paddle/phi/ops/compat/channel_shuffle_sig.cc
index 179098ab06d17..28544f5d1e812 100644
--- a/paddle/phi/ops/compat/channel_shuffle_sig.cc
+++ b/paddle/phi/ops/compat/channel_shuffle_sig.cc
@@ -25,9 +25,9 @@ KernelSignature ChannelShuffleOpArgumentMapping(
 KernelSignature ChannelShuffleGradOpArgumentMapping(
     const ArgumentMappingContext& ctx) {
   return KernelSignature("channel_shuffle_grad",
-                         {GradVarName("Out")},
+                         {"Out@GRAD"},
                          {"groups", "data_format"},
-                         {GradVarName("X")});
+                         {"X@GRAD"});
 }
 
 }  // namespace phi