|
|
|
@ -129,9 +129,8 @@ 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()),
|
|
|
|
|
ctx.device_context());
|
|
|
|
|
return framework::OpKernelType(ctx.Input<framework::Tensor>("X")->type(),
|
|
|
|
|
ctx.device_context());
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
@ -150,9 +149,8 @@ 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()),
|
|
|
|
|
ctx.device_context());
|
|
|
|
|
return framework::OpKernelType(ctx.Input<framework::Tensor>("X")->type(),
|
|
|
|
|
ctx.device_context());
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|