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

test=develop

Share LoD from Input(Rois). (#15420)
test=develop
530869f8 · whs · GitHub · 7ab4af27 · 530869f8
显示空白变更内容
内联并排

Showing with 1 addition and 0 deletion

paddle/fluid/operators/detection/roi_perspective_transform_op.cc ...fluid/operators/detection/roi_perspective_transform_op.cc +1 -0

未找到文件。
--- 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: