diff options
Diffstat (limited to 'tensorflow/tensorflow.bzl')
-rw-r--r-- | tensorflow/tensorflow.bzl | 17 |
1 files changed, 16 insertions, 1 deletions
diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl index cddf16f147..d31b9a6e31 100644 --- a/tensorflow/tensorflow.bzl +++ b/tensorflow/tensorflow.bzl @@ -72,6 +72,13 @@ def if_android_arm64(a): }) +def if_android_mips(a): + return select({ + clean_dep("//tensorflow:android_mips"): a, + "//conditions:default": [], + }) + + def if_not_android(a): return select({ clean_dep("//tensorflow:android"): [], @@ -79,6 +86,14 @@ def if_not_android(a): }) +def if_not_android_mips_and_mips64(a): + return select({ + clean_dep("//tensorflow:android_mips"): [], + clean_dep("//tensorflow:android_mips64"): [], + "//conditions:default": a, + }) + + def if_android(a): return select({ clean_dep("//tensorflow:android"): a, @@ -853,7 +868,7 @@ def cc_header_only_library(name, deps=[], **kwargs): def tf_custom_op_library_additional_deps(): return [ - "@protobuf//:protobuf_headers", + "@protobuf_archive//:protobuf_headers", clean_dep("//third_party/eigen3"), clean_dep("//tensorflow/core:framework_headers_lib"), ] |