Searched refs:send_builder (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_rewrite.cc | 71 auto send_builder = make_builder("_NcclReduceSend", in ReplaceReduce() local
76 TF_RETURN_IF_ERROR(send_builder.Finalize(graph, &send_node)); in ReplaceReduce()
171 NodeBuilder send_builder = make_builder("_NcclBroadcastSend", "Send"); in ReplaceBroadcast() local
174 send_builder.ControlInput(edge->src()); in ReplaceBroadcast()
177 send_builder.Input(in_node); in ReplaceBroadcast()
181 TF_RETURN_IF_ERROR(send_builder.Finalize(graph, &send_node)); in ReplaceBroadcast()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 242 NodeDefBuilder send_builder(opts.new_name(src->name()), send_op, in AddSend() local
244 SetSendRecvAttrs(opts, edge, &send_builder); in AddSend()
245 send_builder.Device(src->assigned_device_name()).Input(send_from); in AddSend()
247 send_builder.Attr("_start_time", start_time); in AddSend()
250 *status = send_builder.Finalize(send, /*consume=*/true); in AddSend()
|