Searched refs:SigNode (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/graph_analyzer/ |
| D | sig_node.h | 38 class SigNode; variable
43 using SigNodeMap = std::map<string, std::unique_ptr<SigNode>>;
53 class SigNode {
59 explicit SigNode(const NodeDef* node);
71 std::unordered_map<const GenNode* /*full_graph*/, SigNode* /*subgraph*/>;
121 using PeerVector = std::vector<SigNode*>;
134 HashedPeer(size_t l, SigNode* p) : link_hash(l), peer(p) {} in HashedPeer()
137 bool operator()(const SigNode::HashedPeer& left, in operator()
138 const SigNode::HashedPeer& right) { in operator()
144 SigNode* peer;
[all …]
|
| D | sig_node_test.cc | 40 SigNode::LinkTag a(GenNode::Port(false, 1), GenNode::Port(false, 2)); in TEST()
41 SigNode::LinkTag b(GenNode::Port(false, 1), GenNode::Port(false, 2)); in TEST()
42 SigNode::LinkTag c(GenNode::Port(false, 2), GenNode::Port(false, 1)); in TEST()
43 SigNode::LinkTag d(GenNode::Port(false, 1), GenNode::Port(false, 3)); in TEST()
44 SigNode::LinkTag e(GenNode::Port(false, 2), GenNode::Port(false, 2)); in TEST()
77 std::map<SigNode::LinkTag, SigNode::Link>* link_map, SigNode* node) { in CopyLinksPass2()
81 static void ComputeTopoHash0(SigNode* node) { node->ComputeTopoHash0(); } in ComputeTopoHash0()
83 static void ComputeTopoHash(int distance, SigNode* node) { in ComputeTopoHash()
87 static size_t GetTopoHash(int distance, SigNode* node) { in GetTopoHash()
91 static size_t GetHighTopoHash(SigNode* node) { in GetHighTopoHash()
[all …]
|
| D | sig_node.cc | 30 SigNode::SigNode(const NodeDef* node) : node_(node) {} in SigNode() function in tensorflow::grappler::graph_analyzer::SigNode
32 void SigNode::CopyLinks(const GenNode& from, const TranslationMap& map) { in CopyLinks()
41 void SigNode::CopyLinksPass1(const GenNode& from, const TranslationMap& map, in CopyLinksPass1()
67 void SigNode::CopyLinksPass2(std::map<LinkTag, Link>* link_map) { in CopyLinksPass2()
88 void SigNode::ComputeTopoHash0() { in ComputeTopoHash0()
104 void SigNode::ComputeTopoHash(int distance) { in ComputeTopoHash()
156 size_t SigNode::GetTopoHash(int distance) const { in GetTopoHash()
167 bool SigNode::operator==(const SigNode& other) const { in operator ==()
260 SigNode* node = entry.second.get(); in PrepareNodes()
279 SigNode::NodeOrderLess()); in FindUniqueHashes()
[all …]
|
| D | test_tools.h | 110 std::vector<string> DumpLinkHashMap(const SigNode::LinkHashMap& link_hash_map);
113 const SigNode::HashedPeerVector& hashed_peers);
|
| D | subgraph.cc | 96 SigNode::TranslationMap full_to_new; in ExtractForSignature()
99 auto newnode_ref = absl::make_unique<SigNode>(node->node_def()); in ExtractForSignature()
|
| D | test_tools.cc | 105 std::vector<string> DumpLinkHashMap(const SigNode::LinkHashMap& link_hash_map) { in DumpLinkHashMap()
108 std::map<SigNode::LinkTag, size_t> tags; in DumpLinkHashMap()
128 const SigNode::HashedPeerVector& hashed_peers) { in DumpHashedPeerVector()
|