From 17212696cd2391dde05455a6003bc291b5cac85e Mon Sep 17 00:00:00 2001 From: ktlichkid Date: Wed, 18 Apr 2018 20:42:34 +0800 Subject: [PATCH] Added BeamSearchOpMaker class --- paddle/fluid/operators/beam_search_op.h | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/paddle/fluid/operators/beam_search_op.h b/paddle/fluid/operators/beam_search_op.h index 0a481a85ce6..11ca9b15c59 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, -- GitLab