未验证 提交 0679feed 编写于 作者: Y yiicy 提交者: GitHub

[ARM] add multiclass_nms2 kernel, test=develop (#2917)

add multiclass_nms2 kernel, test=develop
上级 c0fa7ada
...@@ -429,6 +429,17 @@ REGISTER_LITE_KERNEL(multiclass_nms, ...@@ -429,6 +429,17 @@ REGISTER_LITE_KERNEL(multiclass_nms,
.BindInput("BBoxes", {LiteType::GetTensorTy(TARGET(kHost))}) .BindInput("BBoxes", {LiteType::GetTensorTy(TARGET(kHost))})
.BindInput("Scores", {LiteType::GetTensorTy(TARGET(kHost))}) .BindInput("Scores", {LiteType::GetTensorTy(TARGET(kHost))})
.BindOutput("Out", {LiteType::GetTensorTy(TARGET(kHost))}) .BindOutput("Out", {LiteType::GetTensorTy(TARGET(kHost))})
.Finalize();
REGISTER_LITE_KERNEL(multiclass_nms2,
kHost,
kFloat,
kNCHW,
paddle::lite::kernels::host::MulticlassNmsCompute,
def)
.BindInput("BBoxes", {LiteType::GetTensorTy(TARGET(kHost))})
.BindInput("Scores", {LiteType::GetTensorTy(TARGET(kHost))})
.BindOutput("Out", {LiteType::GetTensorTy(TARGET(kHost))})
.BindOutput("Index", .BindOutput("Index",
{LiteType::GetTensorTy(TARGET(kHost), PRECISION(kInt32))}) {LiteType::GetTensorTy(TARGET(kHost), PRECISION(kInt32))})
.Finalize(); .Finalize();
...@@ -84,3 +84,4 @@ bool MulticlassNmsOpLite::AttachImpl(const cpp::OpDesc& opdesc, ...@@ -84,3 +84,4 @@ bool MulticlassNmsOpLite::AttachImpl(const cpp::OpDesc& opdesc,
} // namespace paddle } // namespace paddle
REGISTER_LITE_OP(multiclass_nms, paddle::lite::operators::MulticlassNmsOpLite); REGISTER_LITE_OP(multiclass_nms, paddle::lite::operators::MulticlassNmsOpLite);
REGISTER_LITE_OP(multiclass_nms2, paddle::lite::operators::MulticlassNmsOpLite);
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册