Searched refs:recv_builder (Results 1 – 2 of 2) sorted by relevance
56 NodeBuilder recv_builder = in ReplaceReduce() local67 recv_builder.Input(src_node); in ReplaceReduce()91 TF_RETURN_IF_ERROR(recv_builder.Finalize(graph, &recv_node)); in ReplaceReduce()
301 NodeDefBuilder recv_builder(opts.new_name(src->name()), recv_op, in AddRecv() local303 SetSendRecvAttrs(opts, edge, &recv_builder); in AddRecv()304 recv_builder.Device(dst->assigned_device_name()) in AddRecv()307 *status = recv_builder.Finalize(recv, /*consume=*/true); in AddRecv()