Fix merge bug

test=develop

Fix merge bug
test=develop
b17444c8 · Yu Yang · 62b5772f · b17444c8
显示空白变更内容
内联并排

Showing with 4 addition and 6 deletion

paddle/fluid/operators/psroi_pool_op.cc paddle/fluid/operators/psroi_pool_op.cc +4 -6

未找到文件。
--- a/paddle/fluid/operators/psroi_pool_op.cc
+++ b/paddle/fluid/operators/psroi_pool_op.cc
@@ -129,8 +129,7 @@ class PSROIPoolOp : public framework::OperatorWithKernel {
 protected:
  framework::OpKernelType GetExpectedKernelType(
      const framework::ExecutionContext& ctx) const override {
-    return framework::OpKernelType(
-        framework::ToDataType(ctx.Input<framework::Tensor>("X")->type()),
+    return framework::OpKernelType(ctx.Input<framework::Tensor>("X")->type(),
                                   ctx.device_context());
  }
 };
@@ -150,8 +149,7 @@ class PSROIPoolGradOp : public framework::OperatorWithKernel {
 protected:
  framework::OpKernelType GetExpectedKernelType(
      const framework::ExecutionContext& ctx) const override {
-    return framework::OpKernelType(
-        framework::ToDataType(ctx.Input<framework::Tensor>("X")->type()),
+    return framework::OpKernelType(ctx.Input<framework::Tensor>("X")->type(),
                                   ctx.device_context());
  }
 };