Searched refs:src_tgt_dims (Results 1 – 1 of 1) sorted by relevance
413 if (auto src_tgt_dims = in ReshardNoCache() local415 return ReshardWithAllToAll(target, *src_tgt_dims); in ReshardNoCache()1295 if (auto src_tgt_dims = GetReshardAllToAllSourceTargetDims( in ReshardPartialReplicateWithAllToAll() local1298 ReshardWithAllToAll(tmp_partial_replicate_sharding, *src_tgt_dims); in ReshardPartialReplicateWithAllToAll()1304 if (auto src_tgt_dims = GetReshardAllToAllSourceTargetDims( in ReshardPartialReplicateWithAllToAll() local1306 return partitioned_hlo.ReshardWithAllToAll(target, *src_tgt_dims); in ReshardPartialReplicateWithAllToAll()