Searched refs:send_builder (Results 1 – 2 of 2) sorted by relevance
71 auto send_builder = make_builder("_NcclReduceSend", in ReplaceReduce() local76 TF_RETURN_IF_ERROR(send_builder.Finalize(graph, &send_node)); in ReplaceReduce()171 NodeBuilder send_builder = make_builder("_NcclBroadcastSend", "Send"); in ReplaceBroadcast() local174 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()
238 NodeDefBuilder send_builder(opts.new_name(src->name()), send_op, in AddSend() local240 SetSendRecvAttrs(opts, edge, &send_builder); in AddSend()241 send_builder.Device(src->assigned_device_name()).Input(send_from); in AddSend()243 send_builder.Attr("_start_time", start_time); in AddSend()246 *status = send_builder.Finalize(send); in AddSend()