Searched refs:recv_node (Results 1 – 3 of 3) 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 | 52 std::size_t operator()(const RecvNodeDescriptor& recv_node) const { in operator ()()
53 return std::hash<const NodeDef*>()(recv_node.node) ^ in operator ()()
54 std::hash<int>()(recv_node.port_num) ^ in operator ()()
55 std::hash<string>()(recv_node.device); in operator ()()
392 RecvNodeDescriptor recv_node(input_node, input_node_port_num, in Init() local
394 auto it = cached_recv_nodes.find(recv_node); in Init()
418 cached_recv_nodes[recv_node] = recv; in Init()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass.cc | 2411 Node* recv_node = subgraph.GetRecvAtHostNode(oc_name); in GetShapeInfoForOutsideCompilationSends() local
2412 if (recv_node != nullptr) { in GetShapeInfoForOutsideCompilationSends()
2413 recv_at_host_names.insert(recv_node->name()); in GetShapeInfoForOutsideCompilationSends()
|