message: Open-sourcing util functions for compute cost analysis and HLO parsing of neural networks. Also open-sourcing hlo_pb2 py_proto as we require it for HLO analysis.
diff: @@ -7,6 +7,7 @@ load(
  "//tensorflow/core/platform:build_config.bzl",
  "tf_proto_library",
  )
  +load("//tensorflow/compiler/xla:xla.bzl", "xla_py_proto_library")
  load("//tensorflow:tensorflow.bzl", "if_libtpu", "tf_cc_test")
  # buildifier: disable=same-origin-load
  @@ -5768,10 +5769,9 @@ tf_cc_test(
  ],
  )
  -# copybara:uncomment_begin(google-only)
  -# py_proto_library(
  -# name = "hlo_pb2",
  -# api_version = 2,
  -# deps = [":hlo_proto"],
  -# )
  -# copybara:uncomment_end
  +xla_py_proto_library(
  + name = "hlo_pb2",
  + api_version = 2,
  + visibility = ["//visibility:public"],
  + deps = [":hlo_proto"],
  +)
  