Merge pull request #50498 from penpornk/cherrypicks_0N2LD

r2.6 cherry-pick request: Add the missing kernels_experimental header file to pip package
This commit is contained in:
Mihai Maruseac 2021-06-28 17:23:36 -07:00 committed by GitHub
commit a5317d67e6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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",