未验证 提交 f2d1cd11 编写于 作者: W wangguanzhong 提交者: GitHub

fix lod level, test=develop (#22755)

上级 79d71234
......@@ -41,6 +41,11 @@ class DistributeFpnProposalsOp : public framework::OperatorWithKernel {
}
ctx->SetOutputsDim("MultiFpnRois", outs_dims);
ctx->SetOutputDim("RestoreIndex", {-1, 1});
if (!ctx->IsRuntime()) {
for (size_t i = 0; i < num_out_rois; ++i) {
ctx->SetLoDLevel("MultiFpnRois", ctx->GetLoDLevel("FpnRois"), i);
}
}
}
protected:
......
......@@ -74,6 +74,9 @@ class MultiClassNMSOp : public framework::OperatorWithKernel {
} else {
ctx->SetOutputDim("Out", {-1, box_dims[2] + 2});
}
if (!ctx->IsRuntime()) {
ctx->SetLoDLevel("Out", std::max(ctx->GetLoDLevel("BBoxes"), 1));
}
}
protected:
......@@ -493,6 +496,9 @@ class MultiClassNMS2Op : public MultiClassNMSOp {
} else {
ctx->SetOutputDim("Index", {-1, 1});
}
if (!ctx->IsRuntime()) {
ctx->SetLoDLevel("Index", std::max(ctx->GetLoDLevel("BBoxes"), 1));
}
}
};
......
......@@ -30,11 +30,8 @@ COMPILE_RUN_OP_WHITE_LIST = [
'rpn_target_assign', \
'retinanet_target_assign', \
'filter_by_instag', \
'multiclass_nms', \
'multiclass_nms2', \
'im2sequence', \
'generate_proposal_labels', \
'distribute_fpn_proposals', \
'detection_map', \
'locality_aware_nms', \
'var_conv_2d'
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册