diff options
Diffstat (limited to 'tensorflow/tools/lib_package/BUILD')
-rw-r--r-- | tensorflow/tools/lib_package/BUILD | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/tensorflow/tools/lib_package/BUILD b/tensorflow/tools/lib_package/BUILD index 9da5d5cb5b..6b2523d52b 100644 --- a/tensorflow/tools/lib_package/BUILD +++ b/tensorflow/tools/lib_package/BUILD @@ -4,6 +4,7 @@ package(default_visibility = ["//visibility:private"]) load("@bazel_tools//tools/build_defs/pkg:pkg.bzl", "pkg_tar") +load("//third_party/mkl:build_defs.bzl", "if_mkl") genrule( name = "libtensorflow_proto", @@ -104,7 +105,10 @@ genrule( "@protobuf//:LICENSE", "@snappy//:COPYING", "@zlib_archive//:zlib.h", - ], + ] + if_mkl([ + "//third_party/mkl:LICENSE", + "@mkl//:LICENSE", + ]), outs = ["include/tensorflow/c/LICENSE"], cmd = "$(location :concat_licenses.sh) $(SRCS) >$@", tools = [":concat_licenses.sh"], @@ -135,7 +139,10 @@ genrule( "@protobuf//:LICENSE", "@snappy//:COPYING", "@zlib_archive//:zlib.h", - ], + ] + if_mkl([ + "//third_party/mkl:LICENSE", + "@mkl//:LICENSE", + ]), outs = ["include/tensorflow/jni/LICENSE"], cmd = "$(location :concat_licenses.sh) $(SRCS) >$@", tools = [":concat_licenses.sh"], |