test=develop

revert-15296-async_double_buffered_py_reader
jerrywgz 6 years ago
parent b10d84bc5a
commit 5246285e34

@ -45,7 +45,7 @@ class BoxClipOp : public framework::OperatorWithKernel {
protected:
framework::OpKernelType GetExpectedKernelType(
const framework::ExecutionContext& ctx) const override {
auto data_type = framework::GetDataTypeOfVar(ctx.InputVar("RpnRois"));
auto data_type = framework::GetDataTypeOfVar(ctx.InputVar("InputBox"));
return framework::OpKernelType(data_type, platform::CPUPlace());
}
};

Loading…
Cancel
Save