Searched defs:target_rank (Results 1 – 3 of 3) sorted by relevance
80 void Permuter::DispatchSend(int src_rank, int target_rank, const Tensor* tensor, in DispatchSend()96 void Permuter::DispatchRecv(int src_rank, int target_rank, Tensor* tensor, in DispatchRecv()
341 int target_rank = send_to_ranks[i]; in RunTree() local
309 int64 target_rank = target_sharding.tile_assignment().num_dimensions() - in PartialReplicateReshardCompatibleSharding() local