Searched refs:sink_node (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_test.cc | 157 Node* sink = graph_.sink_node(); in TEST_F()
187 graph_.AddControlEdge(a, graph_.sink_node()); in TEST_F()
189 graph_.AddControlEdge(c, graph_.sink_node()); in TEST_F()
192 VerifyNodes(a, {graph_.source_node()}, {c, graph_.sink_node()}); in TEST_F()
195 VerifyNodes(c, {a}, {graph_.sink_node()}); in TEST_F()
200 VerifyNodes(graph_.source_node(), {}, {a, graph_.sink_node()}); in TEST_F()
201 VerifyNodes(graph_.sink_node(), {a, c, graph_.source_node()}, {}); in TEST_F()
204 VerifyNodes(a, {}, {c, graph_.sink_node()}); in TEST_F()
205 VerifyNodes(graph_.source_node(), {}, {graph_.sink_node()}); // no more a in TEST_F()
208 VerifyNodes(a, {}, {graph_.sink_node()}); // no more c in TEST_F()
[all …]
|
| D | quantize_training_test.cc | 94 g->AddControlEdge(m1, g->sink_node()); in TEST_F()
145 g->AddControlEdge(m1, g->sink_node()); in TEST_F()
195 g->AddControlEdge(m1, g->sink_node()); in TEST_F()
196 g->AddControlEdge(m2, g->sink_node()); in TEST_F()
207 g->AddControlEdge(backward_m, g->sink_node()); in TEST_F()
249 g->AddControlEdge(m1, g->sink_node()); in TEST_F()
250 g->AddControlEdge(m2, g->sink_node()); in TEST_F()
261 g->AddControlEdge(backward_m, g->sink_node()); in TEST_F()
296 graph->AddControlEdge(matmul, graph->sink_node()); in TEST_F()
333 graph->AddControlEdge(matmul, graph->sink_node()); in TEST_F()
[all …]
|
| D | algorithm.cc | 83 ReverseDFSFrom(g, {g.sink_node()}, enter, leave, stable_comparator); in ReverseDFS()
224 g->AddControlEdge(n, g->sink_node()); in FixupSourceAndSinkEdges()
|
| D | subgraph.cc | 257 g->AddControlEdge(send_node, g->sink_node()); in FetchOutputs()
|
| D | graph.h | 540 Node* sink_node() const { return FindNodeId(kSinkId); } in sink_node() function
|
| D | graph_constructor.cc | 1277 node_map[src.sink_node()] = dest->sink_node(); in CopyGraph()
|
| D | graph.cc | 382 if (source == source_node() || dest == sink_node() || x == kControlSlot || in AddEdge()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | scheduler.cc | 116 return (*asap_times)[graph_->sink_node()->id()]; in ComputeAsap()
141 Node* sinkNode = graph_->sink_node(); in ComputeAlap()
|
| /external/tensorflow/tensorflow/contrib/tensorrt/segment/ |
| D | segment.cc | 98 if (e->dst() == graph->sink_node()) { in ContractEdge()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | constant_folding.cc | 421 node_map[orig_graph->sink_node()] = {constant_graph->sink_node()}; in GetConstantGraph()
|
| D | function.cc | 1468 node_map[src.sink_node()->id()] = dst->sink_node(); in Copy()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass.cc | 1384 (*node_images)[graph_in_->sink_node()] = graph_out->sink_node(); in CopyNodesToOutputGraph()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow_test.cc | 234 graph.AddControlEdge(n, graph.sink_node()); in TEST()
|