Home
last modified time | relevance | path

Searched refs:dest_replica (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcollective_permute_thunk.cc196 for (int64 dest_replica : src_participant.dest_replicas) { in SubmitParticipant() local
198 participants_.at(dest_replica); in SubmitParticipant()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.cc401 for (int dest_replica : p.replica_ids_to_copy_to) { in RunCollectiveOp() local
403 replica_idx_to_participant_idx, dest_replica)]; in RunCollectiveOp()
408 replica_idx_to_participant_idx.erase(dest_replica); in RunCollectiveOp()