diff options
author | Gunhan Gulsoy <gunan@google.com> | 2018-07-29 00:11:42 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-29 00:14:52 -0700 |
commit | cde1c8e835f7fdfbb3ab8a4c8a01edb3a7e569aa (patch) | |
tree | 01bfea2ea0878847ae2617a9beb2aae85d5c9910 /tensorflow/core/platform | |
parent | 2ee2fee11391952ee9e73c66ee99f265d924e15a (diff) |
Add a new kernels option to bazel rules to enable adding dynamic kernel dependencies.
PiperOrigin-RevId: 206473913
Diffstat (limited to 'tensorflow/core/platform')
-rw-r--r-- | tensorflow/core/platform/default/build_config_root.bzl | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/core/platform/default/build_config_root.bzl b/tensorflow/core/platform/default/build_config_root.bzl index 09029a4b25..3a012c23fd 100644 --- a/tensorflow/core/platform/default/build_config_root.bzl +++ b/tensorflow/core/platform/default/build_config_root.bzl @@ -58,3 +58,9 @@ def if_static(extra_deps, otherwise=[]): str(Label("//tensorflow:framework_shared_object")): otherwise, "//conditions:default": extra_deps, }) + +def if_dynamic_kernels(extra_deps, otherwise=[]): + return select({ + str(Label("//tensorflow:dynamic_loaded_kernels")): extra_deps, + "//conditions:default": otherwise, + }) |