experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Internal change.

Browse Source 
PiperOrigin-RevId: 207176147
This commit is contained in:
Jonathan Shen 2018-08-02 15:26:37 -07:00 committed by TensorFlower Gardener
parent a12135ff5f
commit f8afea0e1d
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/tensorflow.bzl
View File

@ -1096,6 +1096,10 @@ def tf_kernel_library(
tf_gpu_kernel_library( tf_gpu_kernel_library(
name=name + "_gpu", srcs=gpu_srcs, deps=deps, **kwargs) name=name + "_gpu", srcs=gpu_srcs, deps=deps, **kwargs)
cuda_deps.extend([":" + name + "_gpu"]) cuda_deps.extend([":" + name + "_gpu"])
kwargs["tags"] = kwargs.get("tags", []) + [
"req_dep=%s" % clean_dep("//tensorflow/core:gpu_lib"),
"req_dep=@local_config_cuda//cuda:cuda_headers",
]
tf_cuda_library( tf_cuda_library(
name=name, name=name,
srcs=srcs, srcs=srcs,