fix share lod, test=develop

0d915078 · jerrywgz · ab9d6a4f · 0d915078
隐藏空白更改
内联并排

Showing with 6 addition and 2 deletion

paddle/fluid/operators/detection/box_coder_op.cc paddle/fluid/operators/detection/box_coder_op.cc +6 -2

未找到文件。
--- 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");
+      }
+    }
  }
 };