From 40a490ccb19593b2e588f951ac58f5f5fffe8b98 Mon Sep 17 00:00:00 2001 From: Henry Tan Date: Wed, 24 Jun 2020 17:50:38 -0700 Subject: [PATCH] TPU library internal refactor. PiperOrigin-RevId: 318177458 Change-Id: I849fb7214fac4eebbfec3a929f030a3de3cbc9a5 --- tensorflow/core/tpu/kernels/tpu_op_util.cc | 1 - tensorflow/stream_executor/tpu/BUILD | 5 +---- 2 files changed, 1 insertion(+), 5 deletions(-) diff --git a/tensorflow/core/tpu/kernels/tpu_op_util.cc b/tensorflow/core/tpu/kernels/tpu_op_util.cc index 31b0cc6c72d..7b7cf468499 100644 --- a/tensorflow/core/tpu/kernels/tpu_op_util.cc +++ b/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& dynamic_shapes, const TPUCompileMetadataProto& metadata, const TpuMeshStateInterface& mesh_state) { diff --git a/tensorflow/stream_executor/tpu/BUILD b/tensorflow/stream_executor/tpu/BUILD index 71c2c728a17..06165e90330 100644 --- a/tensorflow/stream_executor/tpu/BUILD +++ b/tensorflow/stream_executor/tpu/BUILD @@ -1,10 +1,7 @@ # 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 ) -- GitLab