add more ops vjp gen (#57298)

Charles-hit · web-flow · commit ce825f47b78c · 2023-09-14T17:27:35.000+08:00
diff --git a/paddle/fluid/primitive/codegen/gen.py b/paddle/fluid/primitive/codegen/gen.py
@@ -66,6 +66,9 @@
     'cast_grad',
     'slice_double_grad',
     'layer_norm_grad',
+    'embedding_grad',
+    'add_n_grad',
+    'scale_grad',
 ]
 
 
@@ -142,6 +145,10 @@
     'dropout_grad',
     'slice',
     'layer_norm_grad',
+    'embedding_grad',
+    'add_n_grad',
+    'sqrt',
+    'uniform',
 ]
 
 
