未验证 提交 5a611afd 编写于 作者: P Pei Yang 提交者: GitHub

fix trt calib not working bug, test=develop (#21934) (#22110)

上级 34ef38c8
...@@ -18,8 +18,10 @@ ...@@ -18,8 +18,10 @@
#include <gtest/gtest_prod.h> #include <gtest/gtest_prod.h>
#endif #endif
#include <map>
#include <memory> #include <memory>
#include <numeric> #include <numeric>
#include <set>
#include <string> #include <string>
#include <unordered_map> #include <unordered_map>
#include <unordered_set> #include <unordered_set>
...@@ -294,7 +296,7 @@ class GraphPatternDetector { ...@@ -294,7 +296,7 @@ class GraphPatternDetector {
using hit_rcd_t = using hit_rcd_t =
std::pair<Node* /*node in graph*/, PDNode* /*node in pattern*/>; std::pair<Node* /*node in graph*/, PDNode* /*node in pattern*/>;
PDPattern pattern_; PDPattern pattern_;
std::unordered_map<const PDNode*, std::unordered_set<Node*>> pdnodes2nodes_; std::map<const PDNode*, std::set<Node*>> pdnodes2nodes_;
}; };
// some helper methods. // some helper methods.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册