Fix detection_map_op for multi-device. (#8845)

PaddlePaddle · Mar 8, 2018 · ded34b2 · ded34b2
1 parent 7f00716
commit ded34b2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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());
   }
 };