Searched refs:out_node (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_rewrite.cc | 82 for (const auto& out_node : out_nodes) { in ReplaceReduce() local
83 graph->AddControlEdge(send_node, out_node.node); in ReplaceReduce()
94 for (const auto& out_node : out_nodes) { in ReplaceReduce() local
95 if (out_node.index == Graph::kControlSlot) { in ReplaceReduce()
96 graph->AddControlEdge(recv_node, out_node.node); in ReplaceReduce()
98 graph->AddEdge(recv_node, 0, out_node.node, out_node.index); in ReplaceReduce()
150 for (const auto& out_node : out_nodes) { in ReplaceBroadcast() local
151 if (out_node.index == Graph::kControlSlot) { in ReplaceBroadcast()
152 graph->AddControlEdge(in_node, out_node.node); in ReplaceBroadcast()
154 graph->AddEdge(in_node, in_index, out_node.node, out_node.index); in ReplaceBroadcast()
[all …]
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | subgraph.cc | 221 Node** out_node) { in AddNode() argument
232 .Finalize(g, out_node)); in AddNode()
233 (*out_node)->set_assigned_device_name(device_info().name()); in AddNode()
238 Node** out_node) { in AddNode() argument
251 .Finalize(g, out_node)); in AddNode()
253 (*out_node)->set_assigned_device_name(device_info().name()); in AddNode()
258 Node** out_node) { in AddNode() argument
271 .Finalize(g, out_node)); in AddNode()
272 (*out_node)->set_assigned_device_name(device_info().name()); in AddNode()
277 Node** out_node) { in AddNode() argument
[all …]
|
| D | subgraph.h | 60 Node** out_node) = 0;
126 Node** out_node) override;
137 Node** out_node) override;
147 Node** out_node) override;
159 Node** out_node) override;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_execution_state.cc | 259 Node** out_node) override { in AddNode() argument
280 .Finalize(g, out_node)); in AddNode()
282 (*out_node)->set_assigned_device_name( in AddNode()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 3313 NodeDef* out_node = optimized_graph->add_node(); in AddQuantizedMatMulMinMaxOutConstNodes() local
3321 CreateNodeDef(out_const_name, TensorValue(&value), out_node)); in AddQuantizedMatMulMinMaxOutConstNodes()
3322 node_map_->AddNode(out_const_name, out_node); in AddQuantizedMatMulMinMaxOutConstNodes()
3323 out_node->set_device(node->device()); in AddQuantizedMatMulMinMaxOutConstNodes()
3326 out_node->mutable_input()->CopyFrom(node->input()); in AddQuantizedMatMulMinMaxOutConstNodes()
3327 for (const string& input : out_node->input()) { in AddQuantizedMatMulMinMaxOutConstNodes()
|