Skip to content

Commit 423c3bb

Browse files
authored
[fluid_ops] dropout add default value (PaddlePaddle#72220)
* Fix * Fix
1 parent 9d76afa commit 423c3bb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/phi/ops/yaml/ops.yaml

+1-1
Original file line numberDiff line numberDiff line change
@@ -1535,7 +1535,7 @@
15351535
traits : paddle::dialect::ForwardOnlyTrait
15361536

15371537
- op : dropout
1538-
args : (Tensor x, Tensor seed_tensor, Scalar p, bool is_test, str mode, int seed, bool fix_seed)
1538+
args : (Tensor x, Tensor seed_tensor, Scalar p = 0.5f, bool is_test = false, str mode = "downgrade_in_infer", int seed = 0, bool fix_seed = false)
15391539
output : Tensor(out), Tensor(mask)
15401540
infer_meta :
15411541
func : DropoutInferMeta

0 commit comments

Comments
 (0)