diff options
author | Sami Kama <skama@nvidia.com> | 2018-01-29 15:08:26 -0800 |
---|---|---|
committer | Sami Kama <skama@nvidia.com> | 2018-01-29 15:08:26 -0800 |
commit | be50e66950a4534322f55fbcbc300ea710f2b6c6 (patch) | |
tree | e3a70def27b622a97e3b706cb5643371d06b5fff /third_party/tensorrt/tensorrt_configure.bzl | |
parent | ae740a67bdc01b991ead6ac047c774bff4d7bc8f (diff) |
Fix compilation issues with new build files
Diffstat (limited to 'third_party/tensorrt/tensorrt_configure.bzl')
-rw-r--r-- | third_party/tensorrt/tensorrt_configure.bzl | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/third_party/tensorrt/tensorrt_configure.bzl b/third_party/tensorrt/tensorrt_configure.bzl index 8aa0f28f39..4a1441500a 100644 --- a/third_party/tensorrt/tensorrt_configure.bzl +++ b/third_party/tensorrt/tensorrt_configure.bzl @@ -19,9 +19,9 @@ load( _TENSORRT_INSTALL_PATH = "TENSORRT_INSTALL_PATH" _TF_TENSORRT_VERSION = "TF_TENSORRT_VERSION" -_TF_TENSORRT_LIBS = ["nvinfer", "nvinfer_plugin", "nvparsers"] +_TF_TENSORRT_LIBS = ["nvinfer"] _TF_TENSORRT_HEADERS = [ - "NvInfer.h", "NvInferPlugin.h", "NvCaffeParser.h", "NvUffParser.h", + "NvInfer.h", "NvUtils.h" ] |