fix share lod, test=develop

6 years ago · 0d91507859
parent ab9d6a4f39
commit 0d91507859
1 changed files with 6 additions and 2 deletions
--- a/paddle/fluid/operators/detection/box_coder_op.cc
+++ b/paddle/fluid/operators/detection/box_coder_op.cc
@ -77,9 +77,13 @@ class BoxCoderOp : public framework::OperatorWithKernel {
        PADDLE_ENFORCE_EQ(target_box_dims[2], prior_box_dims[1]);
        ctx->ShareDim("TargetBox", /*->*/ "OutputBox");
      }
-    }

-    ctx->ShareLoD("TargetBox", /*->*/ "OutputBox");
+      if (code_type == BoxCodeType::kDecodeCenterSize && axis == 1) {
+        ctx->ShareLoD("PriorBox", /*->*/ "OutputBox");
+      } else {
+        ctx->ShareLoD("TargetBox", /*->*/ "OutputBox");
+      }
+    }
  }
 };