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

Buildify tensorrt's generated BUILD file.

Browse Source 
Ran buildifier and manually fixed the package location.

PiperOrigin-RevId: 240188658
This commit is contained in:
A. Unique TensorFlower 2019-03-25 11:57:10 -07:00 committed by TensorFlower Gardener
parent 97bebc917b
commit f665d09225
1 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
third_party/tensorrt/BUILD.tpl vendored
View File

@ -3,12 +3,12 @@
licenses(["notice"]) licenses(["notice"])
exports_files(["LICENSE"])
load("@local_config_cuda//cuda:build_defs.bzl", "cuda_default_copts") load("@local_config_cuda//cuda:build_defs.bzl", "cuda_default_copts")
package(default_visibility = ["//visibility:public"]) package(default_visibility = ["//visibility:public"])
exports_files(["LICENSE"])
cc_library( cc_library(
name = "tensorrt_headers", name = "tensorrt_headers",
hdrs = [%{tensorrt_headers}], hdrs = [%{tensorrt_headers}],
@ -18,15 +18,14 @@ cc_library(
cc_library( cc_library(
name = "tensorrt", name = "tensorrt",
srcs = %{tensorrt_libs}, srcs = %{tensorrt_libs},
data = %{tensorrt_libs},
copts = cuda_default_copts(), copts = cuda_default_copts(),
deps = [ data = %{tensorrt_libs},
"@local_config_cuda//cuda:cuda",
":tensorrt_headers",
],
linkstatic = 1, linkstatic = 1,
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
deps = [
":tensorrt_headers",
"@local_config_cuda//cuda",
],
) )
%{copy_rules} %{copy_rules}