Added BeamSearchOpMaker class

7 years ago · 17212696cd
parent 9ced6da444
commit 17212696cd
1 changed files with 7 additions and 0 deletions
--- 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,