Function TEST

tensorflow/compiler/xla/service/tuple_util_test.cc:58–69

}

TEST(TupleUtilTest, ExtractPrefix) {
  HloInstruction *param0, *param1;
  HloComputation* entry_computation;

  TF_ASSERT_OK_AND_ASSIGN(
      auto module, GetParsedModule(&entry_computation, &param0, &param1));

  HloInstruction* prefix = TupleUtil::ExtractPrefix(param0, 2);

  EXPECT_THAT(prefix, op::Tuple(op::GetTupleElement(op::Parameter(0), 0),
                                op::GetTupleElement(op::Parameter(0), 1)));
}

TEST(TupleUtilTest, AppendSuffix) {
  HloInstruction *param0, *param1;