[phi] move viterbi_decode to phi (#40186)

* move viterbi to phi * move infershape to phi * update infershape * fix * resolve conflicts

Author: linjieccc

paddle/fluid/operators/viterbi_decode_op.cc

@@ -9,8 +9,10 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License. */
 
-#include "paddle/fluid/operators/viterbi_decode_op.h"
+#include "paddle/fluid/framework/infershape_utils.h"
 #include "paddle/fluid/framework/op_registry.h"
+#include "paddle/phi/core/infermeta_utils.h"
+#include "paddle/phi/infermeta/ternary.h"
 
 namespace paddle {
 namespace operators {
@@ -19,47 +21,6 @@ class ViterbiDecodeOp : public framework::OperatorWithKernel {
  public:
  using framework::OperatorWithKernel::OperatorWithKernel;
 
- void InferShape(framework::InferShapeContext* ctx) const override {
- OP_INOUT_CHECK(ctx->HasInput("Input"), "Input", "Input", "ViterbiDecode");
- OP_INOUT_CHECK(ctx->HasInput("Transition"), "Input", "Transition",
- "ViterbiDecode");
- OP_INOUT_CHECK(ctx->HasInput("Length"), "Input", "Length", "ViterbiDecode");
- OP_INOUT_CHECK(ctx->HasOutput("Scores"), "Output", "Scores",
- "ViterbiDecode");
- OP_INOUT_CHECK(ctx->HasOutput("Path"), "Output", "Path", "ViterbiDecode");
- auto in_dims = ctx->GetInputDim("Input");
- PADDLE_ENFORCE_EQ(in_dims.size(), 3,
- platform::errors::InvalidArgument(
- "The rank of Input in ViterbiDecode must be 3. But "
- "received Input's rank is %d.",
- in_dims.size()));
- auto length_dims = ctx->GetInputDim("Length");
- PADDLE_ENFORCE_EQ(length_dims.size(), 1,
- platform::errors::InvalidArgument(
- "The rank of Length in ViterbiDecode must be 1. But "
- "received Length's rank is %d.",
- length_dims.size()));
- auto transition_dims = ctx->GetInputDim("Transition");
- PADDLE_ENFORCE_EQ(
- transition_dims.size(), 2,
- platform::errors::InvalidArgument(
- "The rank of Transition in ViterbiDecode must be 2. But "
- "received Transition's rank is %d.",
- transition_dims.size()));
- if (ctx->IsRuntime()) {
- PADDLE_ENFORCE_EQ(
- in_dims[0], length_dims[0],
- platform::errors::InvalidArgument(
- "The batch size of Input and Length should be equal."));
- PADDLE_ENFORCE_EQ(in_dims[2], transition_dims[0],
- platform::errors::InvalidArgument(
- "The number of tags of Input (%d) and Transition "
- "(%d) should be equal.",
- transition_dims[0], in_dims[2]));
- }
- ctx->SetOutputDim("Scores", length_dims);
- }
-
  protected:
  framework::OpKernelType GetExpectedKernelType(
  const framework::ExecutionContext& ctx) const override {
@@ -102,8 +63,8 @@ class ViterbiDecodeOpMaker : public framework::OpProtoAndCheckerMaker {
 
 namespace ops = paddle::operators;
 namespace platform = paddle::platform;
+DECLARE_INFER_SHAPE_FUNCTOR(viterbi_decode, ViterbiDecodeInferShapeFunctor,
+ PD_INFER_META(phi::ViterbiDecodeInferMeta));
 REGISTER_OP_WITHOUT_GRADIENT(viterbi_decode, ops::ViterbiDecodeOp,
- ops::ViterbiDecodeOpMaker);
-REGISTER_OP_CPU_KERNEL(
- viterbi_decode, ops::ViterbiDecodeKernel<platform::CPUDeviceContext, float>,
- ops::ViterbiDecodeKernel<platform::CPUDeviceContext, double>);
+ ops::ViterbiDecodeOpMaker,
+ ViterbiDecodeInferShapeFunctor);