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

Internal change

Browse Source 
PiperOrigin-RevId: 201301504
This commit is contained in:
A. Unique TensorFlower 2018-06-19 23:07:57 -07:00 committed by TensorFlower Gardener
parent 081f30a7bc
commit b8f0b7391e
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/tensorflow.bzl
View File

@ -922,6 +922,7 @@ def tf_gpu_kernel_library(srcs,
hdrs=[],
**kwargs):
copts = copts + _cuda_copts() + if_cuda(cuda_copts) + tf_copts()
kwargs["features"] = kwargs.get("features", []) + ["-use_header_modules"]
native.cc_library(
srcs=srcs,
@ -1305,6 +1306,7 @@ def tf_custom_op_library(name, srcs=[], gpu_srcs=[], deps=[], linkopts=[]):
name=basename + "_gpu",
srcs=gpu_srcs,
copts=_cuda_copts() + if_tensorrt(["-DGOOGLE_TENSORRT=1"]),
features = if_cuda(["-use_header_modules"]),
deps=deps + if_cuda(cuda_deps))
cuda_deps.extend([":" + basename + "_gpu"])