提交 f176a9cf 编写于 作者: G guosheng

Remove ElementwiseOpInferVarType in elementwise_op to use the default...

Remove ElementwiseOpInferVarType in elementwise_op to use the default InferVarType to find var recursively
上级 35e55636
...@@ -42,16 +42,6 @@ class ElementwiseOp : public framework::OperatorWithKernel { ...@@ -42,16 +42,6 @@ class ElementwiseOp : public framework::OperatorWithKernel {
} }
}; };
class ElementwiseOpInferVarType : public framework::VarTypeInference {
public:
void operator()(const framework::OpDesc& op_desc,
framework::BlockDesc* block) const override {
auto x_var = op_desc.Input("X")[0];
auto out_var = op_desc.Output("Out")[0];
block->Var(out_var)->SetType(block->Var(x_var)->GetType());
}
};
class ElementwiseOpMaker : public framework::OpProtoAndCheckerMaker { class ElementwiseOpMaker : public framework::OpProtoAndCheckerMaker {
public: public:
void Make() final { void Make() final {
...@@ -148,6 +138,5 @@ class ElementwiseOpGrad : public framework::OperatorWithKernel { ...@@ -148,6 +138,5 @@ class ElementwiseOpGrad : public framework::OperatorWithKernel {
}; \ }; \
REGISTER_OPERATOR(op_type, ::paddle::operators::ElementwiseOp, \ REGISTER_OPERATOR(op_type, ::paddle::operators::ElementwiseOp, \
__ElemwiseOp##op_type##Maker__, \ __ElemwiseOp##op_type##Maker__, \
::paddle::operators::ElementwiseOpInferVarType, \
::paddle::framework::DefaultGradOpDescMaker<true>); \ ::paddle::framework::DefaultGradOpDescMaker<true>); \
REGISTER_OPERATOR(op_type##_grad, ::paddle::operators::ElementwiseOpGrad) REGISTER_OPERATOR(op_type##_grad, ::paddle::operators::ElementwiseOpGrad)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册