Bazel: tensorrt_configure.bzl bugfix

937062f5 · Liu Jiaming · e321b021 · 937062f5
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

tools/tensorrt/tensorrt_configure.bzl tools/tensorrt/tensorrt_configure.bzl +1 -1

未找到文件。
--- a/tools/tensorrt/tensorrt_configure.bzl
+++ b/tools/tensorrt/tensorrt_configure.bzl
@@ -24,7 +24,7 @@ _TF_TENSORRT_CONFIG_REPO = "TF_TENSORRT_CONFIG_REPO"
 _TF_TENSORRT_VERSION = "TF_TENSORRT_VERSION"
 _TF_NEED_TENSORRT = "TF_NEED_TENSORRT"

-_TF_TENSORRT_LIBS = ["nvinfer", "nvinfer_plugin", "nvparsers"]
+_TF_TENSORRT_LIBS = ["nvinfer", "nvinfer_plugin", "nvparsers", "nvonnxparser"]
 _TF_TENSORRT_HEADERS = ["NvInfer.h", "NvUtils.h", "NvInferPlugin.h"]
 _TF_TENSORRT_HEADERS_V6 = [
    "NvInfer.h",