protected:
  // Test auto-clustering with a proto text file ${key}.pbtxt.
  Status RunAutoClusteringTestWithPbtxt(absl::string_view key) {
    string file_name_without_extension =
        absl::StrCat(testing::TensorFlowSrcRoot(), "/compiler/jit/tests/", key);

    return AutoClusteringTest::RunAutoClusteringTestWithPbtxt(
        absl::StrCat(file_name_without_extension, ".pbtxt"),

// rule for :auto_clustering_test and then see the comment on update_golden on
// how to auto-generate the .golden_summary file.

class AutoClusteringTest : public ::testing::Test {
 protected:
  Status RunAutoClusteringTestWithPbtxt(
      absl::string_view pbtxt_file_path,
      absl::string_view golden_summary_file_path);
  Status RunAutoClusteringTestWithGzippedPbtxt(
      absl::string_view gzipped_pbtxt_file_path,

      Env::Default(), string(golden_summary_file_path), &golden_file_contents));

  EXPECT_EQ(golden_file_contents, clustering_summary);

  return absl::OkStatus();
}

Status AutoClusteringTest::RunAutoClusteringTestWithPbtxt(
    absl::string_view pbtxt_file_path,
    absl::string_view golden_summary_file_path) {
  GraphDef graphdef;
  TF_RETURN_IF_ERROR(
      ReadTextProto(Env::Default(), string(pbtxt_file_path), &graphdef));