提交 17212696 编写于 作者: K ktlichkid

Added BeamSearchOpMaker class

上级 9ced6da4
...@@ -192,6 +192,13 @@ std::ostream& operator<<(std::ostream& os, const BeamSearch::Item& item); ...@@ -192,6 +192,13 @@ std::ostream& operator<<(std::ostream& os, const BeamSearch::Item& item);
std::string ItemToString(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 { class BeamSearchOp : public framework::OperatorBase {
public: public:
BeamSearchOp(const std::string& type, BeamSearchOp(const std::string& type,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册