Searched refs:rendez_args (Results 1 – 1 of 1) sorted by relevance
44 Rendezvous::Args rendez_args; in SendTensorsToRendezvous() local45 rendez_args.device_context = device_context; in SendTensorsToRendezvous()47 rendez_args.alloc_attrs = alloc_attrs[i]; in SendTensorsToRendezvous()51 rendezvous->Send(parsed, rendez_args, tensors_to_send[i], false)); in SendTensorsToRendezvous()96 Rendezvous::Args rendez_args; in RecvOutputsFromRendezvousAsync() local97 rendez_args.device_context = device_context; in RecvOutputsFromRendezvousAsync()98 rendez_args.alloc_attrs = std::get<3>(p); in RecvOutputsFromRendezvousAsync()101 parsed, rendez_args, in RecvOutputsFromRendezvousAsync()