-rw-r--r-- | tensorflow/c/BUILD | 5 |
diff --git a/tensorflow/c/BUILD b/tensorflow/c/BUILD index 4cd3e55..db43451 100644 --- a/tensorflow/c/BUILD +++ b/tensorflow/c/BUILD @@ -554,7 +554,10 @@ tf_cuda_library( cc_library( name = "kernels_hdrs", - hdrs = ["kernels.h"], + hdrs = [ + "kernels.h", + "kernels_experimental.h", + ], visibility = ["//tensorflow:internal"], deps = [ ":c_api_internal", |