message: Update data types in all_reduce_simplifier_test for TF2
  Summary:
  REF T48455
  TF2.4 Only
  Reviewers: #tensorflow, #framework_ip_review_-_any_oss_or_third-party_code_use_has_been_approved, babakk
  Maniphest Tasks: T48455
diff: @@ -63,10 +63,11 @@ INSTANTIATE_TEST_SUITE_P(AllReduceSimplifierTestCases, AllReduceSimplifierTest,
  ::testing::Values(AllReduceSimplifierTestSpec{F16},
  AllReduceSimplifierTestSpec{F32},
  AllReduceSimplifierTestSpec{U8},
  + AllReduceSimplifierTestSpec{U16},
  AllReduceSimplifierTestSpec{U32},
  + AllReduceSimplifierTestSpec{S8},
  + AllReduceSimplifierTestSpec{S16},
  AllReduceSimplifierTestSpec{S32}));
  -// Other data types have been disabled due to missing data types in
  -// the implementation of Literal::GetIntegralAsS64() in TF1
  string ReplaceParams(absl::string_view s,
  const AllReduceSimplifierTestSpec& spec) {
  