diff --git a/paddle/fluid/operators/beam_search_op.h b/paddle/fluid/operators/beam_search_op.h index 0a481a85ce6fbb582b8c0e12710455aaaac72aa1..11ca9b15c59a0234420437e0937ab43ab3074c71 100644 --- a/paddle/fluid/operators/beam_search_op.h +++ b/paddle/fluid/operators/beam_search_op.h @@ -192,6 +192,13 @@ std::ostream& operator<<(std::ostream& os, const BeamSearch::Item& item); std::string ItemToString(const BeamSearch::Item& item); +class BeamSearchOpMaker : public framework::OpProtoAndCheckerMaker{ + public: + MulOpMaker(OpProto *proto, OpAttrChecker *op_checker) + : OpProtoAndCheckerMaker(proto, op_checker){ + } +} + class BeamSearchOp : public framework::OperatorBase { public: BeamSearchOp(const std::string& type,