Home
last modified time | relevance | path

Searched refs:rendezvous_key (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_utils.cc218 NcclCliqueRendezvous(const RendezvousKey& rendezvous_key, in NcclCliqueRendezvous() argument
221 : Rendezvous(rendezvous_key), in NcclCliqueRendezvous()
222 key_(std::move(rendezvous_key.global_devices)), in NcclCliqueRendezvous()
330 const RendezvousKey& rendezvous_key, int local_device_ordinal, in AcquireNcclClique() argument
333 VLOG(2) << "Rendezvous key: " << rendezvous_key.ToString() in AcquireNcclClique()
340 NcclCliqueParticipantData participant(rendezvous_key, local_device_ordinal, in AcquireNcclClique()
346 rendezvous_key, [&](const RendezvousKey& rendezvous_key) { in AcquireNcclClique() argument
348 rendezvous_key, local_participants, callback); in AcquireNcclClique()
Dnccl_collective_thunk.cc89 RendezvousKey rendezvous_key(params.run_id, std::move(participants), in ExecuteOnStream() local
97 AcquireNcclClique(rendezvous_key, device_ordinal, params.stream, in ExecuteOnStream()
Dnccl_utils.h151 const RendezvousKey& rendezvous_key, int local_device_ordinal,
/external/tensorflow/tensorflow/compiler/xla/service/
Dcollective_ops_utils.h142 ParticipantData(const RendezvousKey& rendezvous_key, int64 device_ordinal, in ParticipantData()
144 : rendezvous_key(rendezvous_key), in ParticipantData()
150 RendezvousKey rendezvous_key; member
191 absl::StrJoin(buffer_strs, ","), rendezvous_key.ToString(), in ToString()
278 participants_.back().rendezvous_key != participant.rendezvous_key) { in SubmitParticipant()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dhost_compute_ops.cc125 std::vector<string> rendezvous_key(ctx->num_outputs()); in ComputeAsync() local
128 rendezvous_key[i] = Rendezvous::CreateKey( in ComputeAsync()
135 ctx, Rendezvous::ParseKey(rendezvous_key[i], &parsed_key[i]), done); in ComputeAsync()
147 const string& key = rendezvous_key[i]; in ComputeAsync()
275 const string& rendezvous_key = Rendezvous::CreateKey( in Compute() local
282 OP_REQUIRES_OK(ctx, Rendezvous::ParseKey(rendezvous_key, &parsed_key)); in Compute()
283 VLOG(2) << "Send " << rendezvous_key; in Compute()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.cc632 xla::RendezvousKey rendezvous_key = in __xla_cpu_runtime_AllToAll() local
635 AllToAllParticipantData participant(rendezvous_key, device_ordinal, in __xla_cpu_runtime_AllToAll()
653 rendezvous_key, make_cpu_rendezvous); in __xla_cpu_runtime_AllToAll()
670 xla::RendezvousKey rendezvous_key = in __xla_cpu_runtime_AllReduce() local
681 xla::AllReduceParticipantData participant(rendezvous_key, device_ordinal, in __xla_cpu_runtime_AllReduce()
703 rendezvous_key, make_cpu_rendezvous); in __xla_cpu_runtime_AllReduce()
742 xla::RendezvousKey rendezvous_key = in __xla_cpu_runtime_CollectivePermute() local
745 CollectivePermuteParticipantData participant(rendezvous_key, device_ordinal, in __xla_cpu_runtime_CollectivePermute()
760 rendezvous_key, make_cpu_rendezvous); in __xla_cpu_runtime_CollectivePermute()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_worker_service.cc490 const string& key = request->rendezvous_key(); in GrpcRecvTensorAsync()
547 CopyDeviceToHost(&val, alloc, alloc, request->rendezvous_key(), in GrpcRecvTensorAsync()
Dgrpc_remote_worker.cc231 const string& key = request->rendezvous_key(); in RecvTensorAsync()
/external/tensorflow/tensorflow/core/protobuf/
Dworker.proto344 string rendezvous_key = 2; field