message: Tf2.1 -> TF2.4 fixes
  Summary:
  TF2.4 Only
  Ref T31437
  Test Plan: Passes locally
  Reviewers: jackh, #tensorflow, simonl, #framework_ip_review_-_any_oss_or_third-party_code_use_has_been_approved
  Maniphest Tasks: T31437
diff: @@ -255,7 +255,7 @@ class Resnet18_No_Batchnorm(xla_test.XLATestCase):
  sess.run(train, feed_dict={x: data, y_: labels})
  report = pva.openReport(report_helper.find_report())
  - self.assert_total_tile_memory(report, 41595004)
  + self.assert_total_tile_memory(report, 40553247)
  def testTrainingMomentumInLoop(self):
  cfg = ipu.utils.IPUConfig()
  @@ -294,7 +294,7 @@ class Resnet18_No_Batchnorm(xla_test.XLATestCase):
  sess.run(train, feed_dict={x: data, y_: labels})
  report = pva.openReport(report_helper.find_report())
  - self.assert_total_tile_memory(report, 43679590)
  + self.assert_total_tile_memory(report, 42995305)
  def testTrainingInLoopWithGradientAccumulation(self):
  cfg = ipu.utils.IPUConfig()
  @@ -335,7 +335,7 @@ class Resnet18_No_Batchnorm(xla_test.XLATestCase):
  sess.run(train, feed_dict={x: data, y_: labels})
  report = pva.openReport(report_helper.find_report())
  - self.assert_total_tile_memory(report, 45535722)
  + self.assert_total_tile_memory(report, 43369525)
  def testTrainingMomentumInLoopWithGradientAccumulation(self):
  cfg = ipu.utils.IPUConfig()
  @@ -375,7 +375,7 @@ class Resnet18_No_Batchnorm(xla_test.XLATestCase):
  sess.run(train, feed_dict={x: data, y_: labels})
  report = pva.openReport(report_helper.find_report())
  - self.assert_total_tile_memory(report, 45699624)
  + self.assert_total_tile_memory(report, 43636781)
  if __name__ == "__main__":
  