Home
last modified time | relevance | path

Searched refs:out_node (Results 1 – 9 of 9) 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.cc388 for (const auto out_node : switch_node->out_nodes()) { in AddNoOpAfterLastIteration() local
389 if (out_node->IsExit()) { in AddNoOpAfterLastIteration()
390 graph->AddControlEdge(after_last_iteration_node, out_node); in AddNoOpAfterLastIteration()
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.cc319 Node** out_node) override { in AddNode() argument
340 .Finalize(g, out_node)); in AddNode()
342 (*out_node)->set_assigned_device_name( in AddNode()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc3929 NodeDef* out_node = optimized_graph->add_node(); in AddQuantizedMatMulMinMaxOutConstNodes() local
3938 CreateNodeDef(out_const_name, TensorValue(&value), out_node)); in AddQuantizedMatMulMinMaxOutConstNodes()
3939 node_map_->AddNode(out_const_name, out_node); in AddQuantizedMatMulMinMaxOutConstNodes()
3940 out_node->set_device(node->device()); in AddQuantizedMatMulMinMaxOutConstNodes()
3942 out_node->mutable_input()->CopyFrom(node->input()); in AddQuantizedMatMulMinMaxOutConstNodes()
3943 for (const string& input : out_node->input()) { in AddQuantizedMatMulMinMaxOutConstNodes()
/external/tensorflow/tensorflow/dtensor/cc/
Ddtensor_device.cc1192 Node* out_node = out_edge->dst(); in SelectGraphToExecute() local
1193 if (!out_node->IsSink()) nodes_to_remove.push(out_node); in SelectGraphToExecute()
/external/tensorflow/tensorflow/core/framework/
Dmodel.h790 std::shared_ptr<Node> parent, std::shared_ptr<Node>* out_node)
Dmodel.cc2130 std::shared_ptr<Node>* out_node) { in AddNode() argument
2146 *out_node = std::move(node); in AddNode()