| /external/tensorflow/tensorflow/core/grappler/graph_analyzer/ |
| D | gen_node_test.cc | 35 GenNode::Port p(true, 100); in TEST()
39 GenNode::Port p2 = GenNode::Port::Decode(p.Encoded()); in TEST()
45 GenNode::Port p(false, 0); in TEST()
49 GenNode::Port p2 = GenNode::Port::Decode(p.Encoded()); in TEST()
55 GenNode::Port p(true, -100); in TEST()
59 GenNode::Port p2 = GenNode::Port::Decode(p.Encoded()); in TEST()
65 GenNode::Port p(false, -1); in TEST()
69 GenNode::Port p2 = GenNode::Port::Decode(p.Encoded()); in TEST()
79 map["node1"] = absl::make_unique<GenNode>(&node1); in TEST()
91 map["node1"] = absl::make_unique<GenNode>(&node1); in TEST()
[all …]
|
| D | subgraph.h | 36 class Identity : public gtl::FlatSet<const GenNode*> {
38 using InitializerList = std::initializer_list<GenNode*>;
52 Subgraph(const Identity& parent_id, GenNode* add_node);
113 const GenNode::LinkTarget& GetNeighbor() const { in GetNeighbor()
119 const GenNode* GetNode() const { return *id_it_; } in GetNode()
123 GenNode::Port GetPort() const { return link_map_it_->first; } in GetPort()
167 GenNode::LinkMap::const_iterator link_map_it_;
183 Subgraph* ExtendParent(const Subgraph::Identity& parent_id, GenNode* node);
|
| D | gen_node.cc | 28 GenNode::GenNode(const NodeDef* node) : node_(node), op_(nullptr) {} in GenNode() function in tensorflow::grappler::graph_analyzer::GenNode
30 Status GenNode::BuildGraphInMap(const GraphDef& source, GenNodeMap* map) { in BuildGraphInMap()
38 (*map)[name] = absl::make_unique<GenNode>(&n); in BuildGraphInMap()
50 Status GenNode::ParseInputs(const GenNodeMap* map) { in ParseInputs()
102 GenNode* other_node = other_it->second.get(); in ParseInputs()
125 bool GenNode::IsMultiInput(Port port) const { in IsMultiInput()
136 GenNode::Port::operator string() const { in operator string()
|
| D | gen_node.h | 34 class GenNode; variable
37 using GenNodeMap = std::unordered_map<string, std::unique_ptr<GenNode>>;
46 class GenNode {
50 explicit GenNode(const NodeDef* node);
122 GenNode* node; // Node where this link points.
125 LinkTarget(GenNode* a_node, Port a_port) : node(a_node), port(a_port) {} in LinkTarget()
|
| D | sig_node.h | 71 std::unordered_map<const GenNode* /*full_graph*/, SigNode* /*subgraph*/>;
72 void CopyLinks(const GenNode& from, const TranslationMap& map);
89 GenNode::Port::Hasher port_hasher;
92 LinkTag(GenNode::Port a_local, GenNode::Port a_remote) in LinkTag()
101 GenNode::Port local;
103 GenNode::Port remote;
162 void CopyLinksPass1(const GenNode& from, const TranslationMap& map,
|
| D | subgraph_test.cc | 44 ASSERT_THAT(GenNode::BuildGraphInMap(graph, &map), Eq(Status::OK())); in TEST()
67 auto gn1 = absl::make_unique<GenNode>(&node1); in TEST()
91 ASSERT_THAT(GenNode::BuildGraphInMap(graph, &map), Eq(Status::OK())); in TEST()
154 ASSERT_THAT(GenNode::BuildGraphInMap(graph, &map), Eq(Status::OK())); in TEST()
167 GenNode::Port last_port(false, -1); in TEST()
171 GenNode::Port new_port = sit.GetPort(); in TEST()
204 ASSERT_THAT(GenNode::BuildGraphInMap(graph, &map), Eq(Status::OK())); in TEST()
211 const GenNode* last_node = nullptr; in TEST()
214 const GenNode* new_node = sit.GetNode(); in TEST()
255 ASSERT_THAT(GenNode::BuildGraphInMap(graph, &map), Eq(Status::OK())); in TEST()
[all …]
|
| D | graph_analyzer.cc | 61 return GenNode::BuildGraphInMap(graph_, &nodes_); in BuildMap()
99 const GenNode* last_all_or_none_node = nullptr; in ExtendSubgraph()
101 const GenNode* node = sit.GetNode(); in ExtendSubgraph()
102 GenNode::Port port = sit.GetPort(); in ExtendSubgraph()
103 const GenNode::LinkTarget& neighbor = sit.GetNeighbor(); in ExtendSubgraph()
137 const GenNode* node) { in ExtendSubgraphAllOrNone()
166 const GenNode* node, in ExtendSubgraphPortAllOrNone()
167 GenNode::Port port) { in ExtendSubgraphPortAllOrNone()
265 GenNode* peer = sit.GetNeighbor().node; in HasInvalidMultiInputs()
|
| 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()
67 CHECK(GenNode::BuildGraphInMap(graph, &gen_map_).ok()); in BuildSigMap()
147 link1.tag = SigNode::LinkTag(GenNode::Port(true, 0), GenNode::Port(false, 0)); in TEST_F()
152 link2.tag = SigNode::LinkTag(GenNode::Port(true, 1), GenNode::Port(false, 0)); in TEST_F()
157 link3.tag = SigNode::LinkTag(GenNode::Port(true, 2), GenNode::Port(false, 0)); in TEST_F()
482 ASSERT_THAT(GenNode::BuildGraphInMap(graph1, &gen_map1), Eq(Status::OK())); in TEST_F()
[all …]
|
| D | graph_analyzer_test.cc | 48 void ExtendSubgraphPortAllOrNone(Subgraph* parent, GenNode* node, in ExtendSubgraphPortAllOrNone()
49 GenNode::Port port) { in ExtendSubgraphPortAllOrNone()
53 void ExtendSubgraphAllOrNone(Subgraph* parent, GenNode* node) { in ExtendSubgraphAllOrNone()
69 GenNode* GetNode(const string& name) { return gran_->nodes_.at(name).get(); } in GetNode()
157 GenNode::Port(true, 0)); in TEST_F()
180 GenNode::Port(true, 0)); in TEST_F()
202 GenNode::Port(true, 0)); in TEST_F()
224 GenNode::Port(true, 0)); in TEST_F()
243 GenNode::Port(true, 0)); in TEST_F()
261 GenNode::Port(true, 0)); in TEST_F()
[all …]
|
| D | graph_analyzer.h | 90 void ExtendSubgraphAllOrNone(Subgraph* parent, const GenNode* node);
92 void ExtendSubgraphPortAllOrNone(Subgraph* parent, const GenNode* node,
93 GenNode::Port port);
|
| D | subgraph.cc | 71 std::hash<const GenNode*> hasher; in Hash()
112 Subgraph::Subgraph(const Identity& parent_id, GenNode* add_node) in Subgraph()
215 GenNode* node) { in ExtendParent()
|
| D | test_tools.h | 107 std::vector<string> DumpLinkMap(const GenNode::LinkMap& link_map);
|
| D | sig_node.cc | 32 void SigNode::CopyLinks(const GenNode& from, const TranslationMap& map) { in CopyLinks()
41 void SigNode::CopyLinksPass1(const GenNode& from, const TranslationMap& map, in CopyLinksPass1()
|
| D | test_tools.cc | 81 std::vector<string> DumpLinkMap(const GenNode::LinkMap& link_map) { in DumpLinkMap()
|