Home
last modified time | relevance | path

Searched refs:out_node (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/nccl/
Dnccl_rewrite.cc82 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/
Dsubgraph.cc220 Node** out_node) { in AddNode() argument
231 .Finalize(g, out_node, /*consume=*/true)); in AddNode()
232 (*out_node)->set_assigned_device_name(device_info().name()); in AddNode()
237 Node** out_node) { in AddNode() argument
250 .Finalize(g, out_node, /*consume=*/true)); in AddNode()
252 (*out_node)->set_assigned_device_name(device_info().name()); in AddNode()
257 Node** out_node) { in AddNode() argument
270 .Finalize(g, out_node, /*consume=*/true)); in AddNode()
271 (*out_node)->set_assigned_device_name(device_info().name()); in AddNode()
276 Node** out_node) { in AddNode() argument
[all …]
Dsubgraph.h60 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/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.cc351 for (const auto out_node : loop_switch_node->out_nodes()) { in AddNoOpAfterLastIteration() local
352 if (out_node->IsExit()) { in AddNoOpAfterLastIteration()
353 exit_node = out_node; in AddNoOpAfterLastIteration()
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.cc318 Node** out_node) override { in AddNode() argument
339 .Finalize(g, out_node)); in AddNode()
341 (*out_node)->set_assigned_device_name( in AddNode()
/external/tensorflow/tensorflow/core/framework/
Dmodel.h670 std::shared_ptr<Node> parent, std::shared_ptr<Node>* out_node)
Dmodel.cc1599 std::shared_ptr<Node>* out_node) { in AddNode() argument
1617 *out_node = std::move(node); in AddNode()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc3870 NodeDef* out_node = optimized_graph->add_node(); in AddQuantizedMatMulMinMaxOutConstNodes() local
3879 CreateNodeDef(out_const_name, TensorValue(&value), out_node)); in AddQuantizedMatMulMinMaxOutConstNodes()
3880 node_map_->AddNode(out_const_name, out_node); in AddQuantizedMatMulMinMaxOutConstNodes()
3881 out_node->set_device(node->device()); in AddQuantizedMatMulMinMaxOutConstNodes()
3883 out_node->mutable_input()->CopyFrom(node->input()); in AddQuantizedMatMulMinMaxOutConstNodes()
3884 for (const string& input : out_node->input()) { in AddQuantizedMatMulMinMaxOutConstNodes()