Searched refs:try_reshard (Results 1 – 1 of 1) sorted by relevance
419 auto try_reshard = ReshardFromPartialReplicateWithDynamicSlice(target); in ReshardNoCache() local420 if (try_reshard.has_value()) { in ReshardNoCache()421 return try_reshard.value(); in ReshardNoCache()423 try_reshard = ReshardPartialReplicateWithAllToAll(target); in ReshardNoCache()424 if (try_reshard.has_value()) { in ReshardNoCache()425 return try_reshard.value(); in ReshardNoCache()430 auto try_reshard = ReshardToPartialReplicateWithAllGather(target); in ReshardNoCache() local431 if (try_reshard.has_value()) { in ReshardNoCache()432 return try_reshard.value(); in ReshardNoCache()434 try_reshard = ReshardPartialReplicateWithAllToAll(target); in ReshardNoCache()[all …]