[XLA:TPU] Update memory bound loop optimizer tests such that the required memory is the available memory, in other words tighten the tests where more than required memory is available.

[XLA:TPU] Update memory bound loop optimizer tests such that the required memory is the available memory, in other words tighten the tests where more than required memory is available. #51921