Registered beam search op

paddle/fluid/operators/beam_search_op.cc

@@ -316,9 +316,17 @@ class BeamSearchInferVarType : public framework::VarTypeInference {
 */
 }  // namespace operators
 }  // namespace paddle
-
+/*
 REGISTER_OPERATOR(beam_search, paddle::operators::BeamSearchOp,
                   paddle::operators::BeamSearchProtoAndCheckerMaker,
                   paddle::operators::BeamSearchInferShape,
                   paddle::operators::BeamSearchInferVarType,
                   paddle::framework::EmptyGradOpMaker);
+*/
+namespace ops = paddle::operators;
+REGISTER_OP_WITHOUT_GRADIENT(beam_search, ops::BeamSearchOp,
+                             ops::BeamSearchOpMaker);
+REGISTER_OP_CPU_KERNEL(
+    beam_search,
+    ops::BeamSearchOpKernel<paddle::platform::CPUDeviceContext, float>,
+    ops::BeamSearchOpKernel<paddle::platform::CPUDeviceContext, double>);

paddle/fluid/operators/beam_search_op.h

@@ -193,7 +193,7 @@ std::ostream& operator<<(std::ostream& os, const BeamSearch::Item& item);
 std::string ItemToString(const BeamSearch::Item& item);
 
 template <typename DeviceContext, typename T>
-class BeamSearchKernel : public framework::OpKernel<T>{
+class BeamSearchOpKernel : public framework::OpKernel<T>{
  public:
   void Compute(const framework::ExecutionContext& context) const override {
     auto* ids_var = context.Input<framework::Tensor>("ids");