From 530869f829821347405e7a80ee5e07c9bb5daa94 Mon Sep 17 00:00:00 2001 From: whs Date: Mon, 21 Jan 2019 10:44:43 +0800 Subject: [PATCH] Share LoD from Input(Rois). (#15420) test=develop --- paddle/fluid/operators/detection/roi_perspective_transform_op.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/paddle/fluid/operators/detection/roi_perspective_transform_op.cc b/paddle/fluid/operators/detection/roi_perspective_transform_op.cc index 3796854fe67..a97828e6fe9 100644 --- a/paddle/fluid/operators/detection/roi_perspective_transform_op.cc +++ b/paddle/fluid/operators/detection/roi_perspective_transform_op.cc @@ -493,6 +493,7 @@ class ROIPerspectiveTransformOp : public framework::OperatorWithKernel { auto out_dims = framework::make_ddim(out_dims_v); ctx->SetOutputDim("Out", out_dims); + ctx->ShareLoD("ROIs", /*->*/ "Out"); } protected: -- GitLab