message: Adjust bounds
  Summary: Ref T16509
  Test Plan: All tests pass in V1
  Reviewers: #tensorflow, #framework_ip_review_-_any_oss_or_third-party_code_use_has_been_approved, davidn
  Subscribers: davidn
  Maniphest Tasks: T16509
diff: @@ -159,7 +159,7 @@ class Resnet18_No_Batchnorm(xla_test.XLATestCase):
  sess.run(train, feed_dict={x: data, y_: labels})
  report.parse_log()
  - report.assert_total_tile_memory(46556154)
  + report.assert_total_tile_memory(44460761)
  def testTrainingMomentum(self):
  with self.session() as sess:
  @@ -187,7 +187,7 @@ class Resnet18_No_Batchnorm(xla_test.XLATestCase):
  sess.run(train, feed_dict={x: data, y_: labels})
  report.parse_log()
  - report.assert_total_tile_memory(49209135)
  + report.assert_total_tile_memory(46632692)
  def testTrainingInLoop(self):
  with self.session() as sess:
  @@ -223,7 +223,7 @@ class Resnet18_No_Batchnorm(xla_test.XLATestCase):
  sess.run(train, feed_dict={x: data, y_: labels})
  report.parse_log()
  - report.assert_total_tile_memory(48167468)
  + report.assert_total_tile_memory(45971962)
  def testTrainingMomentumInLoop(self):
  with self.session() as sess:
  @@ -258,7 +258,7 @@ class Resnet18_No_Batchnorm(xla_test.XLATestCase):
  sess.run(train, feed_dict={x: data, y_: labels})
  report.parse_log()
  - report.assert_total_tile_memory(50370016)
  + report.assert_total_tile_memory(48045630)
  def testTrainingInLoopWithGradientAccumulation(self):
  with self.session() as sess:
  @@ -295,7 +295,7 @@ class Resnet18_No_Batchnorm(xla_test.XLATestCase):
  sess.run(train, feed_dict={x: data, y_: labels})
  report.parse_log()
  - report.assert_total_tile_memory(50517852)
  + report.assert_total_tile_memory(46698896)
  def testTrainingMomentumInLoopWithGradientAccumulation(self):
  with self.session() as sess:
  