Home
last modified time | relevance | path

Searched refs:recv_builder (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/nccl/
Dnccl_rewrite.cc56 NodeBuilder recv_builder = in ReplaceReduce() local
67 recv_builder.Input(src_node); in ReplaceReduce()
91 TF_RETURN_IF_ERROR(recv_builder.Finalize(graph, &recv_node)); in ReplaceReduce()
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition.cc274 NodeDefBuilder recv_builder(opts.new_name(src->name()), recv_op, in AddRecv() local
276 SetSendRecvAttrs(opts, edge, &recv_builder); in AddRecv()
277 recv_builder.Device(dst->assigned_device_name()) in AddRecv()
280 *status = recv_builder.Finalize(recv); in AddRecv()