Adopts the syntax 'xla_tpu_auto_spmd_partitioning_memory_budget_ratio=-1.2' where the new '-' sign indicates that we are disabling the soft memory constraint.
Adopts the syntax 'xla_tpu_auto_spmd_partitioning_memory_budget_ratio=-1.2' where the new '-' sign indicates that we are disabling the soft memory constraint.
