Update pixel_unshuffle_sig.cc

PaddlePaddle · Apr 18, 2022 · c3fbce6 · c3fbce6
1 parent 9aeb8b8
commit c3fbce6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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