Fix detection_map_op for multi-device. (#8845)

7 years ago · ded34b2c0f
parent 7f00716c87
commit ded34b2c0f
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/operators/detection_map_op.cc
+++ b/paddle/fluid/operators/detection_map_op.cc
@ -71,7 +71,7 @@ class DetectionMAPOp : public framework::OperatorWithKernel {
    return framework::OpKernelType(
        framework::ToDataType(
            ctx.Input<framework::Tensor>("DetectRes")->type()),
-        ctx.device_context());
+        platform::CPUPlace());
  }
 };