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

add op_function_generator for box_coder (#28303)

* add op_function_generator for box_coder

* fix format
上级 e1fb4673
...@@ -53,6 +53,7 @@ std::map<std::string, std::set<std::string>> op_ins_map = { ...@@ -53,6 +53,7 @@ std::map<std::string, std::set<std::string>> op_ins_map = {
{"X", "W", "Label", "PathTable", "PathCode", "Bias"}}, {"X", "W", "Label", "PathTable", "PathCode", "Bias"}},
{"moving_average_abs_max_scale", {"X", "InAccum", "InState"}}, {"moving_average_abs_max_scale", {"X", "InAccum", "InState"}},
{"multiclass_nms3", {"BBoxes", "Scores", "RoisNum"}}, {"multiclass_nms3", {"BBoxes", "Scores", "RoisNum"}},
{"box_coder", {"PriorBox", "PriorBoxVar", "TargetBox"}},
}; };
// NOTE(zhiqiu): Like op_ins_map. // NOTE(zhiqiu): Like op_ins_map.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册