Moving TPU ops components as TPU kernels library.

PiperOrigin-RevId: 312578499 Change-Id: I10d7675bfe4a79b65008ffd8f21c6f807dda266e
2020-05-20 16:33:24 -07:00 · 2020-05-20 16:33:24 -07:00 · ae14cc6b1b
commit ae14cc6b1b
parent eb6dddfbb2
1 changed files with 1 additions and 0 deletions
--- a/tensorflow/compiler/jit/BUILD
+++ b/tensorflow/compiler/jit/BUILD
@ -505,6 +505,7 @@ cc_library(
    name = "shape_inference",
    srcs = ["shape_inference.cc"],
    hdrs = ["shape_inference.h"],
+    visibility = [":friends"],
    deps = [
        ":shape_inference_helpers",
        "//tensorflow/compiler/xla:statusor",