message: Adjust tests
  Summary: Fix T31877.
  Test Plan: CI
  Reviewers: #tensorflow, #framework_ip_review_-_any_oss_or_third-party_code_use_has_been_approved, georgep
  Maniphest Tasks: T31877
diff: @@ -1345,8 +1345,8 @@ class InfeedOutfeedTest(test_util.TensorFlowTestCase):
  self.assertAllClose(outfed, expected)
  report.parse_log()
  - report.assert_max_tile_memory(53276)
  - report.assert_total_tile_memory(58365790)
  + report.assert_max_tile_memory(50236)
  + report.assert_total_tile_memory(56082270)
  total_outfeeds = 0
  for s in report.get_execution_reports()[0]['simulation']['steps']:
  