HomeSort by relevance Sort by last modified time
    Searched defs:recv_node (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/nccl/
nccl_rewrite.cc 90 Node* recv_node = nullptr; local
91 TF_RETURN_IF_ERROR(recv_builder.Finalize(graph, &recv_node));
92 recv_node->set_assigned_device_name_index(recv_dev);
96 graph->AddControlEdge(recv_node, out_node.node);
98 graph->AddEdge(recv_node, 0, out_node.node, out_node.index);
228 Node* recv_node; local
232 .Finalize(graph, &recv_node));
233 recv_node->set_assigned_device_name_index(recv_dev);
235 graph->AddEdge(recv_node, 0, out_node.node, out_node.index);
  /external/tensorflow/tensorflow/compiler/jit/
encapsulate_subgraphs_pass.cc 2411 Node* recv_node = subgraph.GetRecvAtHostNode(oc_name); local
    [all...]

Completed in 127 milliseconds