提交 91cc5d62 编写于 作者: C caoying03

add the forward operator.

上级 515981d7
...@@ -119,7 +119,7 @@ class LinearChainCrfOp : public framework::OperatorWithKernel { ...@@ -119,7 +119,7 @@ class LinearChainCrfOp : public framework::OperatorWithKernel {
using framework::OperatorWithKernel::OperatorWithKernel; using framework::OperatorWithKernel::OperatorWithKernel;
protected: protected:
void InferShape(framework::InferShapeContextBase* ctx) const override {} void InferShape(framework::InferShapeContext* ctx) const override {}
}; };
class LinearChainCrfGradOp : public framework::OperatorWithKernel { class LinearChainCrfGradOp : public framework::OperatorWithKernel {
...@@ -127,7 +127,7 @@ class LinearChainCrfGradOp : public framework::OperatorWithKernel { ...@@ -127,7 +127,7 @@ class LinearChainCrfGradOp : public framework::OperatorWithKernel {
using framework::OperatorWithKernel::OperatorWithKernel; using framework::OperatorWithKernel::OperatorWithKernel;
protected: protected:
void InferShape(framework::InferShapeContextBase* ctx) const override {} void InferShape(framework::InferShapeContext* ctx) const override {}
}; };
} // namespace operators } // namespace operators
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册