diff --git a/paddle/fluid/operators/fused/fused_dropout_helper.h b/paddle/fluid/operators/fused/fused_dropout_helper.h index 33fde64164d1293b5716e595609b4dde7fa4160f..970b2d82e2b15c7c2856db4abb5a8f448f5b0ed5 100644 --- a/paddle/fluid/operators/fused/fused_dropout_helper.h +++ b/paddle/fluid/operators/fused/fused_dropout_helper.h @@ -51,6 +51,18 @@ struct DropoutParam { seed_val = 0; } + DropoutParam(bool fix_seed_, uint64_t seed_, bool is_test_, + bool is_upscale_in_train_, float dropout_prob_, + const framework::Tensor* tensor_seed_, int seed_val_) { + fix_seed = fix_seed_; + seed = seed_; + is_test = is_test_; + is_upscale_in_train = is_upscale_in_train_; + dropout_prob = dropout_prob_; + tensor_seed = tensor_seed_; + seed_val = seed_val_; + } + /** * dropout_index: can be 0, 1, 2. 0 means there is only one dropout, * 1 and 2 represent two dropout, the parameter name of dropout