TPU library internal refactor.

PiperOrigin-RevId: 318177458
Change-Id: I849fb7214fac4eebbfec3a929f030a3de3cbc9a5

tensorflow/core/tpu/kernels/tpu_op_util.cc

@@ -157,7 +157,6 @@ TpuCompilationCacheKey CreateCompilationCacheKey(
 TpuCompilationCacheKey CreateCompilationCacheKey(
     absl::string_view function_name, uint64 function_library_fingerprint,
     absl::string_view mlir_module, const OpInputList& guaranteed_constants,
-    size_t guaranteed_constants_size,
     const std::vector<TensorShape>& dynamic_shapes,
     const TPUCompileMetadataProto& metadata,
     const TpuMeshStateInterface& mesh_state) {

tensorflow/stream_executor/tpu/BUILD

 # Description: StreamExecutor Interface for TPUs
 
 package(
-    default_visibility = [
-        "//learning/brain/tfrc/runtime/tpu_driver:__subpackages__",
-        "//tensorflow/core/tpu:__subpackages__",
-    ],
+    default_visibility = ["//tensorflow/core/tpu:__subpackages__"],
     licenses = ["notice"],  # Apache 2.0
 )