未验证 提交 3e5008ad 编写于 作者: P Pei Yang 提交者: GitHub

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

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