diff --git a/paddle/phi/api/yaml/op_compat.yaml b/paddle/phi/api/yaml/op_compat.yaml
index d5520e4e2a7dc051fc43c0359dd20f738e29fb71..0a36b4d6de995d1394d75f0bb2fbda36b588ebf4 100755
--- a/paddle/phi/api/yaml/op_compat.yaml
+++ b/paddle/phi/api/yaml/op_compat.yaml
@@ -117,7 +117,7 @@
   inputs :
     input : Theta
   outputs :
-    out : Output
+    output : Output
   int_array:
     output_shape :
       data_type : int
diff --git a/paddle/phi/api/yaml/ops.yaml b/paddle/phi/api/yaml/ops.yaml
index 27d65100172cb42c82bcdd68a068321096e08bad..b4082b55da5f734d99874b3449a8432e485b7b94 100644
--- a/paddle/phi/api/yaml/ops.yaml
+++ b/paddle/phi/api/yaml/ops.yaml
@@ -100,7 +100,7 @@
 
 - op : affine_grid
   args : (Tensor input, IntArray output_shape={}, bool align_corners=true)
-  output : Tensor
+  output : Tensor(output)
   infer_meta :
     func : AffineGridInferMeta
     param : [input, output_shape, align_corners]