message: Update bounds
  Summary: Ref T16509
  Test Plan: all tests pass
  Reviewers: jakeh, #tensorflow, #framework_ip_review_-_any_oss_or_third-party_code_use_has_been_approved, hakons
  Maniphest Tasks: T16509
diff: @@ -179,8 +179,8 @@ class MatMulSizeTest(xla_test.XLATestCase):
  for k in (x, y, z)})
  self.assertAllClose(np.full([B, I], 129.0), output[0])
  report.parse_log()
  - report.assert_total_tile_memory(138912857)
  - report.assert_max_tile_memory(114711)
  + report.assert_total_tile_memory(125580981)
  + report.assert_max_tile_memory(108555)
  if __name__ == "__main__":
  