Searched refs:may_reshard_with_allreduce (Results 1 – 1 of 1) sorted by relevance
537 [&](bool may_reshard_with_allreduce) -> StatusOr<HloInstruction*> { in PartitionBaseCase() argument542 if (!may_reshard_with_allreduce && in PartitionBaseCase()559 if (!may_reshard_with_allreduce && in PartitionBaseCase()