fix in backward.yaml

PaddlePaddle · Apr 2, 2024 · c7efe6e · c7efe6e
1 parent cff13c9
commit c7efe6e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/phi/api/yaml/backward.yaml b/paddle/phi/api/yaml/backward.yaml
@@ -2524,7 +2524,7 @@
 
 - backward_op : thresholded_relu_grad
   forward : thresholded_relu (Tensor x, float threshold, float value) -> Tensor(out)
-  args : (Tensor x, Tensor out_grad, float threshold)
+  args : (Tensor x, Tensor out_grad, float threshold, float value)
   output : Tensor(x_grad)
   infer_meta :
     func : UnchangedInferMeta