diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-11-01 15:47:29 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-01 15:50:30 -0700 |
| commit | 36a4b6c815559a583da093a5c19bce5494f6f66d (patch) | |
| tree | 5ea957bce5a0d296a18a7a04c13fda3434dd8e56 /tensorflow/tensorflow.bzl | |
| parent | 6df1854646d9cc442ba109d3af15c2978d1599ac (diff) | |
n/a (internal change only)
PiperOrigin-RevId: 174248361
Diffstat (limited to 'tensorflow/tensorflow.bzl')
| -rw-r--r-- | tensorflow/tensorflow.bzl | 19 |
1 files changed, 18 insertions, 1 deletions
diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl index e647a78055..9e6bb8d710 100644 --- a/tensorflow/tensorflow.bzl +++ b/tensorflow/tensorflow.bzl @@ -276,6 +276,10 @@ def tf_cc_shared_object( }), **kwargs) +register_extension_info( + extension_name="tf_cc_shared_object", + label_regex_for_dep="{extension_name}") + # Links in the framework shared object # (//third_party/tensorflow:libtensorflow_framework.so) when not building @@ -590,6 +594,10 @@ def tf_cc_test_gpu(name, suffix=suffix, args=args) +register_extension_info( + extension_name="tf_cc_test_gpu", + label_regex_for_dep="{extension_name}") + def tf_cuda_cc_test(name, srcs=[], @@ -630,6 +638,11 @@ def tf_cuda_cc_test(name, linkopts=linkopts, args=args) +register_extension_info( + extension_name="tf_cuda_cc_test", + label_regex_for_dep="{extension_name}") + + def tf_cuda_only_cc_test(name, srcs=[], deps=[], @@ -1173,6 +1186,10 @@ def tf_custom_op_library(name, srcs=[], gpu_srcs=[], deps=[]): clean_dep("//tensorflow:darwin"): [], }),) +register_extension_info( + extension_name="tf_custom_op_library", + label_regex_for_dep="{extension_name}") + def tf_custom_op_py_library(name, srcs=[], @@ -1319,7 +1336,7 @@ def tf_py_test(name, register_extension_info( extension_name="tf_py_test", - label_regex_map={"deps": "additional_deps:{extension_name}"}) + label_regex_map={"additional_deps": "deps:{extension_name}"}) def cuda_py_test(name, |