Searched refs:recv_node (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_rewrite.cc | 90 Node* recv_node = nullptr; in ReplaceReduce() local
91 TF_RETURN_IF_ERROR(recv_builder.Finalize(graph, &recv_node)); in ReplaceReduce()
92 recv_node->set_assigned_device_name_index(recv_dev); 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()
228 Node* recv_node; in ReplaceBroadcast() local
232 .Finalize(graph, &recv_node)); in ReplaceBroadcast()
233 recv_node->set_assigned_device_name_index(recv_dev); in ReplaceBroadcast()
235 graph->AddEdge(recv_node, 0, out_node.node, out_node.index); in ReplaceBroadcast()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler.cc | 79 std::size_t operator()(const RecvNodeDescriptor& recv_node) const { in operator ()()
80 return std::hash<const NodeDef*>()(recv_node.node) ^ in operator ()()
81 std::hash<int>()(recv_node.port_num) ^ in operator ()()
82 std::hash<string>()(recv_node.device); in operator ()()
513 RecvNodeDescriptor recv_node(input_node, input_node_port_num, in Init() local
515 auto it = cached_recv_nodes.find(recv_node); in Init()
540 cached_recv_nodes[recv_node] = recv; in Init()
|