Share LoD from Input(Rois). (#15420)

test=develop
6 years ago · 530869f829
parent 7ab4af2716
commit 530869f829
1 changed files with 1 additions and 0 deletions
--- 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: