diff options
| author |  Yifei Feng <yifeif@google.com> | 2018-02-22 14:24:57 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-22 14:29:27 -0800 |
| commit | dce9a49c19f406ba45919e8c94474e55dc5ccd54 (patch) | |
| tree | 928db8a52603e00aef76985cda16b8bceb9debb2 /third_party/tensorrt/BUILD.tpl | |
| parent | cb7e1963c625fd9713e7475d85621f95be6762f1 (diff) | |
Merge changes from github.
PiperOrigin-RevId: 186674197
Diffstat (limited to 'third_party/tensorrt/BUILD.tpl')
| -rw-r--r-- | third_party/tensorrt/BUILD.tpl | 34 |
1 files changed, 3 insertions, 31 deletions
diff --git a/third_party/tensorrt/BUILD.tpl b/third_party/tensorrt/BUILD.tpl index feaeb0bea6..57682e8735 100644 --- a/third_party/tensorrt/BUILD.tpl +++ b/third_party/tensorrt/BUILD.tpl @@ -3,6 +3,8 @@ licenses(["notice"]) +exports_files(["LICENSE"]) + load("@local_config_cuda//cuda:build_defs.bzl", "cuda_default_copts") package(default_visibility = ["//visibility:public"]) @@ -32,36 +34,6 @@ cc_library( visibility = ["//visibility:public"], ) -cc_library( - name = "nv_infer_plugin", - srcs = [%{nv_infer_plugin}], - data = [%{nv_infer_plugin}], - includes = [ - "include", - ], - copts= cuda_default_copts(), - deps = [ - "@local_config_cuda//cuda:cuda", - ":nv_infer", - ":tensorrt_headers", - ], - linkstatic = 1, - visibility = ["//visibility:public"], -) - -cc_library( - name = "nv_parsers", - srcs = [%{nv_parsers}], - data = [%{nv_parsers}], - includes = [ - "include", - ], - copts= cuda_default_copts(), - deps = [ - ":tensorrt_headers", - ], - linkstatic = 1, - visibility = ["//visibility:public"], -) %{tensorrt_genrules} + |