| /external/tensorflow/tensorflow/core/graph/ |
| graph_test.cc | 157 Node* sink = graph_.sink_node();
187 graph_.AddControlEdge(a, graph_.sink_node());
189 graph_.AddControlEdge(c, graph_.sink_node());
192 VerifyNodes(a, {graph_.source_node()}, {c, graph_.sink_node()});
195 VerifyNodes(c, {a}, {graph_.sink_node()});
200 VerifyNodes(graph_.source_node(), {}, {a, graph_.sink_node()});
201 VerifyNodes(graph_.sink_node(), {a, c, graph_.source_node()}, {});
204 VerifyNodes(a, {}, {c, graph_.sink_node()});
205 VerifyNodes(graph_.source_node(), {}, {graph_.sink_node()}); // no more a
208 VerifyNodes(a, {}, {graph_.sink_node()}); // no more
[all...] |
| quantize_training_test.cc | 94 g->AddControlEdge(m1, g->sink_node());
145 g->AddControlEdge(m1, g->sink_node());
195 g->AddControlEdge(m1, g->sink_node());
196 g->AddControlEdge(m2, g->sink_node());
207 g->AddControlEdge(backward_m, g->sink_node());
249 g->AddControlEdge(m1, g->sink_node());
250 g->AddControlEdge(m2, g->sink_node());
261 g->AddControlEdge(backward_m, g->sink_node());
296 graph->AddControlEdge(matmul, graph->sink_node());
333 graph->AddControlEdge(matmul, graph->sink_node());
[all...] |
| algorithm.cc | 83 ReverseDFSFrom(g, {g.sink_node()}, enter, leave, stable_comparator);
224 g->AddControlEdge(n, g->sink_node());
|
| subgraph.cc | 257 g->AddControlEdge(send_node, g->sink_node());
|
| graph.h | 540 Node* sink_node() const { return FindNodeId(kSinkId); } function in class:tensorflow::Graph
|
| graph_constructor.cc | [all...] |
| graph.cc | 382 if (source == source_node() || dest == sink_node() || x == kControlSlot ||
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| scheduler.cc | 116 return (*asap_times)[graph_->sink_node()->id()];
141 Node* sinkNode = graph_->sink_node();
|
| /external/tensorflow/tensorflow/contrib/tensorrt/segment/ |
| segment.cc | 98 if (e->dst() == graph->sink_node()) {
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| constant_folding.cc | 421 node_map[orig_graph->sink_node()] = {constant_graph->sink_node()};
|
| function.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/jit/ |
| encapsulate_subgraphs_pass.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| functionalize_control_flow_test.cc | 234 graph.AddControlEdge(n, graph.sink_node());
[all...] |
