Comment out a number of google-internal targets when copybara-exporting instead of removing them.

PiperOrigin-RevId: 353848826
Change-Id: I0801c0e713a0c63597deb5aed31c8bdb37999c6a

tensorflow/BUILD

@@ -121,6 +121,29 @@ config_setting(
     visibility = ["//visibility:public"],
 )
 
+# copybara:uncomment_begin(google-only)
+# config_setting(
+#     name = "chromiumos_x86_64",
+#     flag_values = {"//tools/cpp:cc_target_os": "chromiumos"},
+#     values = {"cpu": "k8"},
+#     visibility = ["//visibility:public"],
+# )
+#
+# config_setting(
+#     name = "chromiumos_arm64",
+#     flag_values = {"//tools/cpp:cc_target_os": "chromiumos"},
+#     values = {"cpu": "arm"},
+#     visibility = ["//visibility:public"],
+# )
+#
+# config_setting(
+#     name = "chromiumos_armv7",
+#     flag_values = {"//tools/cpp:cc_target_os": "chromiumos"},
+#     values = {"cpu": "armeabi-v7a"},
+#     visibility = ["//visibility:public"],
+# )
+# copybara:uncomment_end
+
 config_setting(
     name = "emscripten",
     values = {"crosstool_top": "//external:android/emscripten"},
@@ -584,6 +607,17 @@ selects.config_setting_group(
     ],
 )
 
+# This flag disables all google production dependencies, intended for
+# applications run with non-prod environment.
+# TODO(timshen): Currently this option only disables some dependencies.
+# See b/122528503.
+# copybara:uncomment_begin(google-only)
+# config_setting(
+#     name = "no_prod_deps",
+#     define_values = {"tf_no_prod_deps": "1"},
+# )
+# copybara:uncomment_end
+
 config_setting(
     name = "lite_protos_legacy",
     define_values = {"TENSORFLOW_PROTOS": "lite"},
@@ -617,6 +651,14 @@ selects.config_setting_group(
     ],
 )
 
+# copybara:uncomment_begin(google-only)
+# config_setting(
+#     name = "portable_proto_force_third_party",
+#     define_values = {"PORTABLE_PROTO_TRANSITION_MODE": "third_party"},
+#     visibility = ["//visibility:public"],
+# )
+# copybara:uncomment_end
+
 # 'enable_registration_v2' opts-in to a different implementation of op and
 # kernel registration - REGISTER_OP, REGISTER_KERNEL_BUILDER, etc.
 #

tensorflow/cc/saved_model/BUILD

@@ -38,6 +38,14 @@ cc_library(
     hdrs = ["tag_constants.h"],
 )
 
+# copybara:uncomment_begin(google-only)
+# cc_library(
+#     name = "mobile_only_deps",
+#     visibility = ["//visibility:private"],
+#     deps = if_mobile(["//tensorflow/core:portable_tensorflow_lib"]),
+# )
+# copybara:uncomment_end
+
 cc_library(
     name = "reader",
     srcs = ["reader.cc"],

tensorflow/compiler/tf2tensorrt/BUILD

@@ -624,3 +624,12 @@ pybind_extension(
         "@pybind11",
     ],
 )
+
+# copybara:uncomment_begin(google-only)
+# py_proto_library(
+#     name = "trt_engine_instance_proto_py_pb2",
+#     has_services = 0,
+#     api_version = 2,
+#     deps = [":trt_engine_instance_proto"],
+# )
+# copybara:uncomment_end

tensorflow/compiler/xla/BUILD

@@ -972,6 +972,15 @@ cc_library(
 
 # -----------------------------------------------------------------------------
 
+# copybara:uncomment_begin(google-only)
+# py_proto_library(
+#     name = "xla_data_proto_py_pb2",
+#     api_version = 2,
+#     visibility = [":friends"],
+#     deps = [":xla_data_proto"],
+# )
+# copybara:uncomment_end
+
 # This is a headers target that extra XLA devices can use to prevent circular dependencies.  Devices that are compiled as separate shared objects can also use it to prevent linking of library code.
 cc_header_only_library(
     name = "xla_headers_lib",

tensorflow/compiler/xla/rpc/BUILD

@@ -103,3 +103,19 @@ cc_library(
         tf_grpc_cc_dependency(),
     ],
 )
+
+# copybara:uncomment_begin(google-only)
+# py_proto_library(
+#     name = "xla_py_pb2",
+#     has_services = True,
+#     api_version = 2,
+#     visibility = ["//visibility:public"],
+#     deps = [":xla_service_proto"],
+# )
+#
+# xla_py_grpc_library(
+#     name = "xla_py_pb2_grpc",
+#     srcs = [":xla_service_proto"],
+#     deps = [":xla_py_pb2"],
+# )
+# copybara:uncomment_end

tensorflow/compiler/xla/service/BUILD

@@ -5377,3 +5377,11 @@ tf_cc_test(
         "@com_google_absl//absl/strings",
     ],
 )
+
+# copybara:uncomment_begin(google-only)
+# py_proto_library(
+#     name = "hlo_pb2",
+#     api_version = 2,
+#     deps = [":hlo_proto"],
+# )
+# copybara:uncomment_end

tensorflow/compiler/xrt/BUILD

@@ -143,3 +143,12 @@ cc_library(
         "//tensorflow/compiler/xrt/kernels:xrt_ops",
     ],
 )
+
+# copybara:uncomment_begin(google-only)
+# py_proto_library(
+#     name = "xrt_proto_py_pb2",
+#     api_version = 2,
+#     visibility = ["//visibility:public"],
+#     deps = [":xrt_proto"],
+# )
+# copybara:uncomment_end

tensorflow/core/BUILD

@@ -1047,6 +1047,26 @@ cc_library(
     alwayslink = 1,
 )
 
+# copybara:uncomment_begin(google-only)
+# cc_library(
+#     name = "portable_tensorflow_lib_lite_no_runtime",
+#     srcs = if_mobile([":mobile_srcs_no_runtime"]),
+#     copts = tf_copts(android_optimization_level_override = None) + tf_opts_nortti_if_lite_protos() + if_ios(["-Os"]),
+#     defines = ["SUPPORT_SELECTIVE_REGISTRATION"] + if_chromiumos(["IS_MOBILE_PLATFORM"]) + tf_defines_nortti_if_lite_protos(),
+#     tags = [
+#         "manual",
+#         "notap",
+#     ],
+#     visibility = ["//visibility:public"],
+#     deps = [
+#         "//tensorflow/core/util:abstract_stack_trace",
+#         "//tensorflow/core/util:stats_calculator_portable",
+#         "//tensorflow/core:mobile_additional_lib_deps",
+#     ] + tf_portable_proto_lib() + tf_portable_deps_no_runtime(),
+#     alwayslink = 1,
+# )
+# copybara:uncomment_end
+
 cc_library(
     name = "mobile_additional_lib_deps",
     deps = tf_additional_lib_deps() + [
@@ -1485,6 +1505,13 @@ alias(
     actual = "//tensorflow/core/protobuf:error_codes_proto_impl_cc",
 )
 
+# copybara:uncomment_begin(google-only)
+# alias(
+#     name = "error_codes_proto",
+#     actual = "//tensorflow/core/lib/core:error_codes_proto",
+# )
+# copybara:uncomment_end
+
 alias(
     name = "version_lib",
     actual = "//tensorflow/core/util:version_info",
@@ -2008,3 +2035,39 @@ transitive_hdrs(
         "//tensorflow/core/platform:stream_executor",
     ],
 )
+
+# copybara:uncomment_begin(google-only)
+# py_proto_library(
+#     name = "protos_all_py_pb2",
+#     has_services = 0,
+#     api_version = 2,
+#     visibility = ["//visibility:public"],
+#     deps = [":protos_all"],
+# )
+#
+# alias(
+#     name = "worker_service_cc_grpc_proto",
+#     actual = "//tensorflow/core/protobuf:worker_service_cc_grpc_proto",
+#     visibility = ["//tensorflow:internal"],
+# )
+#
+# alias(
+#     name = "master_service_cc_grpc_proto",
+#     actual = "//tensorflow/core/protobuf:master_service_cc_grpc_proto",
+#     visibility = ["//tensorflow:internal"],
+# )
+#
+# go_library(
+#     name = "protos_all_go_proto",
+#     srcs = ["protos_all_go_proto.go"],
+#     visibility = ["//visibility:public"],
+#     deps = [
+#         "//tensorflow/core/example:example_parser_configuration_go_proto",
+#         "//tensorflow/core/example:example_protos_go_proto",
+#         "//tensorflow/core/protobuf:for_core_protos_go_proto",
+#         "//tensorflow/core/util:event_go_proto",
+#         "//tensorflow/core/util:memmapped_file_system_go_proto",
+#         "//tensorflow/core/util:saved_tensor_slice_go_proto",
+#     ],
+# )
+# copybara:uncomment_end

tensorflow/core/debug/BUILD

@@ -293,6 +293,21 @@ tf_cc_binary(
     ],
 )
 
+# copybara:uncomment_begin(google-only)
+# py_proto_library(
+#     name = "debug_service_py_pb2",
+#     has_services = 1,
+#     api_version = 2,
+#     deps = [":debug_service_proto"],
+# )
+#
+# py_proto_library(
+#     name = "debugger_event_metadata_py_pb2",
+#     api_version = 2,
+#     deps = [":debugger_event_metadata_proto"],
+# )
+# copybara:uncomment_end
+
 # TODO(cais): Add the following back in when tfdbg is supported on Android.
 # filegroup(
 #     name = "android_srcs",

tensorflow/core/grappler/costs/BUILD

@@ -390,3 +390,13 @@ tf_cc_test(
         "//tensorflow/core/grappler/clusters:virtual_cluster",
     ],
 )
+
+# copybara:uncomment_begin(google-only)
+# py_proto_library(
+#     name = "op_performance_data_py_pb2",
+#     has_services = 0,
+#     api_version = 2,
+#     visibility = ["//visibility:public"],
+#     deps = [":op_performance_data"],
+# )
+# copybara:uncomment_end

tensorflow/core/kernels/boosted_trees/BUILD

@@ -139,3 +139,13 @@ tf_kernel_library(
         ":training_ops",
     ],
 )
+
+# copybara:uncomment_begin(google-only)
+# py_proto_library(
+#     name = "boosted_trees_py_pb2",
+#     has_services = 0,
+#     api_version = 2,
+#     visibility = ["//visibility:public"],
+#     deps = [":boosted_trees_proto"],
+# )
+# copybara:uncomment_end

tensorflow/core/profiler/BUILD

@@ -1,3 +1,4 @@
+# copybara:uncomment(oss-unused) load("//net/grpc/go/build_defs:go_grpc_library.bzl", "go_grpc_library")
 load("//tensorflow:tensorflow.bzl", "filegroup")
 load("//tensorflow:tensorflow.bzl", "tf_cc_binary")
 load("//tensorflow/core/platform:rules_cc.bzl", "cc_library")
@@ -140,3 +141,20 @@ filegroup(
     ],
     visibility = ["//tensorflow/core:__pkg__"],
 )
+
+# copybara:uncomment_begin(google-only)
+# py_proto_library(
+#     name = "profiler_analysis_proto_py_pb2",
+#     has_services = 1,
+#     api_version = 2,
+#     visibility = ["//visibility:public"],
+#     deps = [":profiler_analysis_proto"],
+# )
+#
+# py_proto_library(
+#     name = "protos_all_py_pb2",
+#     api_version = 2,
+#     visibility = [":friends"],
+#     deps = [":protos_all"],
+# )
+# copybara:uncomment_end

tensorflow/core/profiler/protobuf/BUILD

@@ -176,3 +176,19 @@ tf_proto_library(
     cc_api_version = 2,
     visibility = [":friends"],
 )
+
+# copybara:uncomment_begin(google-only)
+# py_proto_library(
+#     name = "xplane_py_pb2",
+#     api_version = 2,
+#     visibility = [":friends"],
+#     deps = [":xplane_proto"],
+# )
+#
+# py_proto_library(
+#     name = "trace_events_py_pb2",
+#     api_version = 2,
+#     visibility = ["//visibility:public"],
+#     deps = [":trace_events_proto"],
+# )
+# copybara:uncomment_end

tensorflow/core/protobuf/BUILD

@@ -1,3 +1,9 @@
+# copybara:uncomment_begin(oss-unused)
+# load("//net/grpc/go/build_defs:go_grpc_library.bzl", "go_grpc_library")
+# load("//net/grpc/python:py_grpc_library.bzl", "py_grpc_library")
+# load("//net/grpc:cc_grpc_library.bzl", "cc_grpc_library")
+# copybara:uncomment_end
+
 # For platform specific build config
 load(
     "//tensorflow/core/platform:build_config.bzl",
@@ -190,3 +196,22 @@ tf_proto_library(
         "//tensorflow/core/framework:protos_all",
     ],
 )
+
+# copybara:uncomment_begin(google-only)
+# cc_grpc_library(
+#     name = "worker_service_cc_grpc_proto",
+#     srcs = [":worker_service_proto"],
+#     generate_mocks = True,
+#     service_namespace = "grpc_gen",
+#     deps = [":worker_service_proto_cc"],
+# )
+#
+# cc_grpc_library(
+#     name = "master_service_cc_grpc_proto",
+#     srcs = [":master_service_proto"],
+#     compatible_with = ["//buildenv/target:gce"],
+#     generate_mocks = True,
+#     service_namespace = "grpc_gen",
+#     deps = [":master_service_proto_cc"],
+# )
+# copybara:uncomment_end

tensorflow/core/protobuf/tpu/BUILD

@@ -92,3 +92,54 @@ tf_pyclif_proto_library(
     proto_srcfile = "topology.proto",
     visibility = ["//visibility:public"],
 )
+
+# copybara:uncomment_begin(google-only)
+# py_proto_library(
+#     name = "tpu_embedding_configuration_py_pb2",
+#     api_version = 2,
+#     visibility = ["//visibility:public"],
+#     deps = [":tpu_embedding_configuration_proto"],
+# )
+#
+# py_proto_library(
+#     name = "optimization_parameters_py_pb2",
+#     api_version = 2,
+#     visibility = ["//visibility:public"],
+#     deps = [":optimization_parameters_proto"],
+# )
+#
+# py_proto_library(
+#     name = "tpu_embedding_output_layout_py_pb2",
+#     api_version = 2,
+#     visibility = ["//visibility:public"],
+#     deps = [":tpu_embedding_output_layout_proto"],
+# )
+#
+# py_proto_library(
+#     name = "topology_py_pb2",
+#     api_version = 2,
+#     visibility = ["//visibility:public"],
+#     deps = [":topology_proto"],
+# )
+#
+# py_proto_library(
+#     name = "dynamic_padding_py_pb2",
+#     api_version = 2,
+#     visibility = ["//visibility:public"],
+#     deps = [":dynamic_padding_proto"],
+# )
+#
+# py_proto_library(
+#     name = "compilation_result_py_pb2",
+#     api_version = 2,
+#     visibility = ["//visibility:public"],
+#     deps = [":compilation_result_proto"],
+# )
+#
+# py_proto_library(
+#     name = "compile_metadata_py_pb2",
+#     api_version = 2,
+#     visibility = ["//visibility:public"],
+#     deps = [":compile_metadata_proto"],
+# )
+# copybara:uncomment_end

tensorflow/lite/testing/BUILD

@@ -1,3 +1,4 @@
+# copybara:uncomment(oss-unused) load("//tools/build_defs/build_test:build_test.bzl", "build_test")
 load(
     "//tensorflow/lite:build_def.bzl",
     "gen_zip_test",
@@ -645,11 +646,18 @@ edgetpu_ops = [
     "transpose_conv",
 ]
 
-[gen_zipped_test_file(
-    name = "zip_%s_edgetpu" % op_name,
-    file = "%s_edgetpu.zip" % op_name,
-    flags = " --make_edgetpu_tests",
-    toco = "//tensorflow/lite/toco:toco",  # Unused
-) for op_name in edgetpu_ops]
-
-edgetpu_targets = [":zip_%s_edgetpu" % op_name for op_name in edgetpu_ops]
+# copybara:uncomment_begin(google-only)
+# [gen_zipped_test_file(
+#     name = "zip_%s_edgetpu" % op_name,
+#     file = "%s_edgetpu.zip" % op_name,
+#     flags = " --make_edgetpu_tests",
+#     toco = "//tensorflow/lite/toco:toco",  # Unused
+# ) for op_name in edgetpu_ops]
+#
+# edgetpu_targets = [":zip_%s_edgetpu" % op_name for op_name in edgetpu_ops]
+#
+# build_test(
+#     name = "gen_edgetpu_tests",
+#     targets = edgetpu_targets,
+# )
+# copybara:uncomment_end

tensorflow/lite/toco/BUILD

@@ -488,3 +488,19 @@ tf_cc_test(
         "@com_google_googletest//:gtest",
     ],
 )
+
+# copybara:uncomment_begin(google-only)
+# py_proto_library(
+#     name = "model_flags_proto_py",
+#     api_version = 2,
+#     visibility = ["//visibility:public"],
+#     deps = [":model_flags_proto"],
+# )
+#
+# py_proto_library(
+#     name = "toco_flags_proto_py",
+#     api_version = 2,
+#     visibility = ["//visibility:public"],
+#     deps = [":toco_flags_proto"],
+# )
+# copybara:uncomment_end

tensorflow/lite/toco/logging/BUILD

@@ -89,3 +89,12 @@ py_test(
         "//tensorflow/python:client_testlib",
     ],
 )
+
+# copybara:uncomment_begin(google-only)
+# py_proto_library(
+#     name = "toco_conversion_log_proto_py",
+#     api_version = 2,
+#     visibility = ["//visibility:public"],
+#     deps = [":toco_conversion_log_proto"],
+# )
+# copybara:uncomment_end

tensorflow/lite/tools/evaluation/proto/BUILD

@@ -85,3 +85,12 @@ cc_proto_library(
     name = "preprocessing_steps_cc_proto",
     deps = ["preprocessing_steps_proto"],
 )
+
+# copybara:uncomment_begin(google-only)
+# py_proto_library(
+#     name = "evaluation_stages_py_pb2",
+#     api_version = 2,
+#     visibility = ["//visibility:public"],
+#     deps = [":evaluation_stages_proto"],
+# )
+# copybara:uncomment_end

tensorflow/python/BUILD

@@ -7000,3 +7000,19 @@ py_library(
     srcs_version = "PY3",
     deps = ["//tensorflow/python/keras/optimizer_v2:legacy_learning_rate_decay"],
 )
+
+# copybara:uncomment_begin(google-only)
+# py_proto_library(
+#     name = "protos_all_py_pb2",
+#     has_services = 0,
+#     api_version = 2,
+#     deps = [":protos_all"],
+# )
+#
+# py_proto_library(
+#     name = "cpp_shape_inference_proto_py_pb2",
+#     has_services = 0,
+#     api_version = 2,
+#     deps = [":cpp_shape_inference_proto"],
+# )
+# copybara:uncomment_end

tensorflow/python/distribute/cluster_resolver/BUILD

@@ -185,3 +185,21 @@ tf_py_test(
         "//tensorflow/python:training_server_lib",
     ],
 )
+
+# copybara:uncomment_begin(google-only)
+# tf_py_test(
+#     name = "brain_jobs_cluster_resolver_test",
+#     size = "small",
+#     srcs = ["brain_jobs_cluster_resolver_test.py"],
+#     grpc_enabled = True,
+#     main = "brain_jobs_cluster_resolver_test.py",
+#     deps = [
+#         ":brain_jobs_cluster_resolver_py",
+#         "//tensorflow/python:client_testlib",
+#         "//tensorflow/python:framework_for_generated_wrappers",
+#         "//tensorflow/python:framework_test_lib",
+#         "//tensorflow/python:platform_test",
+#         "//tensorflow/python:training_server_lib",
+#     ],
+# )
+# copybara:uncomment_end

tensorflow/python/keras/BUILD

@@ -326,6 +326,17 @@ py_library(
     ],
 )
 
+# Internally six.moves.urllib.request.urlretrieve library requires Google
+# SSL context to be provided to work in python 3. This isn't needed in OSS.
+# copybara:uncomment_begin(google-only)
+# py_library(
+#     name = "url_utils",
+#     srcs = ["google/url_utils.py"],
+#     srcs_version = "PY3",
+#     deps = ["//pyglib/contrib/google_ssl"],
+# )
+# copybara:uncomment_end
+
 py_library(
     name = "testing_utils",
     srcs = [
@@ -660,3 +671,29 @@ tf_py_test(
         "@absl_py//absl/testing:parameterized",
     ],
 )
+
+# copybara:uncomment_begin(google-only)
+# tf_py_test(
+#     name = "url_utils_test",
+#     srcs = ["google/url_utils_test.py"],
+#     python_version = "PY3",
+#     deps = [
+#         ":url_utils",
+#         "//testing/pymocks:matchers",
+#         "//tensorflow/python:client_testlib",
+#     ],
+# )
+#
+# tf_py_test(
+#     name = "gfile_utils_test",
+#     size = "small",
+#     srcs = ["google_utils/gfile_utils_test.py"],
+#     python_version = "PY3",
+#     deps = [
+#         ":keras",
+#         "//testing/pymocks:matchers",
+#         "//tensorflow/python:client_testlib",
+#         "//tensorflow/python:util",
+#     ],
+# )
+# copybara:uncomment_end

tensorflow/python/keras/protobuf/BUILD

@@ -21,3 +21,17 @@ tf_proto_library(
     cc_api_version = 2,
     protodeps = ["//tensorflow/core:protos_all"],
 )
+
+# copybara:uncomment_begin(google-only)
+# py_proto_library(
+#     name = "saved_metadata_proto_py_pb2",
+#     api_version = 2,
+#     deps = [":saved_metadata_proto"],
+# )
+#
+# py_proto_library(
+#     name = "projector_config_proto_py_pb2",
+#     api_version = 2,
+#     deps = [":projector_config_proto"],
+# )
+# copybara:uncomment_end

tensorflow/python/kernel_tests/proto/BUILD

@@ -139,3 +139,11 @@ tf_py_test(
         "//tensorflow/python:proto_ops",
     ],
 )
+
+# copybara:uncomment_begin(google-only)
+# py_proto_library(
+#     name = "test_example_proto_py",
+#     api_version = 2,
+#     deps = [":test_example_proto"],
+# )
+# copybara:uncomment_end

tensorflow/python/tools/BUILD

@@ -515,3 +515,10 @@ tf_cc_test(
         "//tensorflow/core/platform:logging",
     ]),
 )
+
+# copybara:uncomment_begin(google-only)
+# gensignature(
+#     name = "inspect_checkpoint.par_sig",
+#     srcs = [":inspect_checkpoint.par"],
+# )
+# copybara:uncomment_end

tensorflow/python/tpu/BUILD

@@ -712,3 +712,12 @@ tf_proto_library(
     ],
     visibility = ["//visibility:public"],
 )
+
+# copybara:uncomment_begin(google-only)
+# py_proto_library(
+#     name = "tensor_tracer_py_pb2",
+#     api_version = 2,
+#     visibility = ["//visibility:public"],
+#     deps = [":tensor_tracer_proto"],
+# )
+# copybara:uncomment_end

tensorflow/python/util/BUILD

@@ -660,3 +660,13 @@ filegroup(
     name = "compare_test_proto_src",
     srcs = ["protobuf/compare_test.proto"],
 )
+
+# copybara:uncomment_begin(google-only)
+# py_proto_library(
+#     name = "compare_test_py_pb2",
+#     testonly = 1,
+#     has_services = 0,
+#     api_version = 2,
+#     deps = [":compare_test_proto"],
+# )
+# copybara:uncomment_end

tensorflow/tools/api/lib/BUILD

@@ -26,3 +26,13 @@ py_library(
         "@six_archive//:six",
     ],
 )
+
+# copybara:uncomment_begin(google-only)
+# py_proto_library(
+#     name = "api_objects_proto_py_pb2",
+#     has_services = 0,
+#     api_version = 2,
+#     visibility = ["//visibility:public"],
+#     deps = [":api_objects_proto"],
+# )
+# copybara:uncomment_end