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

release/1.7
Pei Yang 6 years ago committed by GitHub
parent 15e41e9115
commit 3e5008ad01
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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.

Loading…
Cancel
Save