diff options
author | 2018-06-18 15:52:14 -0700 | |
---|---|---|
committer | 2018-06-18 15:55:08 -0700 | |
commit | f91b5b0896e3ed2b57a32b5a21068b9b5c55899e (patch) | |
tree | 7cecf8ec57db08476692dafd17d36560690281db /tensorflow/tensorflow.bzl | |
parent | 3029a930c4f6e2ca3eadfb75bf25068645e055aa (diff) |
Internal change.
PiperOrigin-RevId: 201073792
Diffstat (limited to 'tensorflow/tensorflow.bzl')
-rw-r--r-- | tensorflow/tensorflow.bzl | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl index 522965990b..1f9fbad0b4 100644 --- a/tensorflow/tensorflow.bzl +++ b/tensorflow/tensorflow.bzl @@ -241,6 +241,9 @@ def tf_opts_nortti_if_android(): # LINT.ThenChange(//tensorflow/contrib/android/cmake/CMakeLists.txt) +def tf_features_nomodules_if_android(): + return if_android(["-use_header_modules"]) + # Given a list of "op_lib_names" (a list of files in the ops directory # without their .cc extensions), generate a library for that file. def tf_gen_op_libs(op_lib_names, deps=None, is_external=True): @@ -959,6 +962,7 @@ def tf_cuda_library(deps=None, cuda_deps=None, copts=tf_copts(), **kwargs): if not cuda_deps: cuda_deps = [] + kwargs["features"] = kwargs.get("features", []) + ["-use_header_modules"] native.cc_library( deps=deps + if_cuda(cuda_deps + [ clean_dep("//tensorflow/core:cuda"), |