| /external/tensorflow/third_party/toolchains/preconfig/ubuntu16.04/tensorrt6.0/ |
| D | BUILD | 16 "tensorrt/include/tensorrt_config.h",
19 include_prefix = "third_party/tensorrt",
20 strip_include_prefix = "tensorrt/include",
24 name = "tensorrt",
46 "tensorrt/lib/libnvinfer.so.6",
47 "tensorrt/lib/libnvinfer_plugin.so.6",
49 …cmd = """cp -f "/usr/lib/x86_64-linux-gnu/libnvinfer.so.6" "$(location tensorrt/lib/libnvinfer.so.…
50 cp -f "/usr/lib/x86_64-linux-gnu/libnvinfer_plugin.so.6" "$(location tensorrt/lib/libnvinfer_plugin…
56 "tensorrt/include/NvInfer.h",
57 "tensorrt/include/NvUtils.h",
[all …]
|
| /external/tensorflow/third_party/toolchains/preconfig/centos6/tensorrt5/ |
| D | BUILD | 15 "tensorrt/include/tensorrt_config.h",
18 include_prefix = "third_party/tensorrt",
19 strip_include_prefix = "tensorrt/include",
23 name = "tensorrt",
45 "tensorrt/lib/libnvinfer.so.5",
46 "tensorrt/lib/libnvinfer_plugin.so.5",
48 cmd = """cp -f "/usr/lib64/libnvinfer.so.5" "$(location tensorrt/lib/libnvinfer.so.5)" && \
49 cp -f "/usr/lib64/libnvinfer_plugin.so.5" "$(location tensorrt/lib/libnvinfer_plugin.so.5)" """,
55 "tensorrt/include/NvInfer.h",
56 "tensorrt/include/NvUtils.h",
[all …]
|
| /external/tensorflow/third_party/toolchains/preconfig/ubuntu16.04/tensorrt5/ |
| D | BUILD | 16 "tensorrt/include/tensorrt_config.h",
19 include_prefix = "third_party/tensorrt",
20 strip_include_prefix = "tensorrt/include",
24 name = "tensorrt",
46 "tensorrt/lib/libnvinfer.so.5",
47 "tensorrt/lib/libnvinfer_plugin.so.5",
49 …cmd = """cp -f "/usr/lib/x86_64-linux-gnu/libnvinfer.so.5" "$(location tensorrt/lib/libnvinfer.so.…
50 cp -f "/usr/lib/x86_64-linux-gnu/libnvinfer_plugin.so.5" "$(location tensorrt/lib/libnvinfer_plugin…
56 "tensorrt/include/NvInfer.h",
57 "tensorrt/include/NvUtils.h",
[all …]
|
| /external/tensorflow/third_party/toolchains/preconfig/ubuntu16.04/tensorrt5.1/ |
| D | BUILD | 16 "tensorrt/include/tensorrt_config.h",
19 include_prefix = "third_party/tensorrt",
20 strip_include_prefix = "tensorrt/include",
24 name = "tensorrt",
46 "tensorrt/lib/libnvinfer.so.5",
47 "tensorrt/lib/libnvinfer_plugin.so.5",
49 …cmd = """cp -f "/usr/lib/x86_64-linux-gnu/libnvinfer.so.5" "$(location tensorrt/lib/libnvinfer.so.…
50 cp -f "/usr/lib/x86_64-linux-gnu/libnvinfer_plugin.so.5" "$(location tensorrt/lib/libnvinfer_plugin…
56 "tensorrt/include/NvInfer.h",
57 "tensorrt/include/NvUtils.h",
[all …]
|
| /external/tensorflow/third_party/toolchains/preconfig/ubuntu14.04/tensorrt5/ |
| D | BUILD | 16 "tensorrt/include/tensorrt_config.h",
19 include_prefix = "third_party/tensorrt",
20 strip_include_prefix = "tensorrt/include",
24 name = "tensorrt",
46 "tensorrt/lib/libnvinfer.so.5",
47 "tensorrt/lib/libnvinfer_plugin.so.5",
49 …cmd = """cp -f "/usr/lib/x86_64-linux-gnu/libnvinfer.so.5" "$(location tensorrt/lib/libnvinfer.so.…
50 cp -f "/usr/lib/x86_64-linux-gnu/libnvinfer_plugin.so.5" "$(location tensorrt/lib/libnvinfer_plugin…
56 "tensorrt/include/NvInfer.h",
57 "tensorrt/include/NvUtils.h",
[all …]
|
| /external/tensorflow/third_party/tensorrt/ |
| D | tensorrt_configure.bzl | 51 Label("//third_party/tensorrt:%s.tpl" % tpl),
63 _tpl(repository_ctx, "tensorrt/include/tensorrt_config.h", {
83 "tensorrt/include/tensorrt_config.h",
84 Label(remote_config_repo + ":tensorrt/include/tensorrt_config.h"),
97 Label("//third_party/tensorrt:LICENSE"),
105 config = find_cuda_config(repository_ctx, ["tensorrt"])
119 outs = ["tensorrt/lib/" + library for library in libraries],
125 outs = ["tensorrt/include/" + header for header in headers],
139 _tpl(repository_ctx, "tensorrt/include/tensorrt_config.h", {
|
| D | BUILD.tpl | 16 "tensorrt/include/tensorrt_config.h",
19 include_prefix = "third_party/tensorrt",
20 strip_include_prefix = "tensorrt/include",
24 name = "tensorrt",
|
| /external/tensorflow/tensorflow/tools/ci_build/ |
| D | Dockerfile.rbe.cuda10.0-cudnn7-centos6 | 7 # 1. Download tensorrt from
8 # https://developer.nvidia.com/nvidia-tensorrt-5x-download
14 # tensorrt for centos is only availble via a authenticated download link from
26 # TODO(klimek): Once nvidia provides a way to install tensorrt from within
27 # docker, switch to that. Until then, we copy the tensorrt rpm into the docker
29 ADD tensorrt.rpm tensorrt.rpm
30 RUN rpm -ihv tensorrt.rpm && \
31 yum install -y --nogpgcheck tensorrt && \
32 rm -f tensorrt.rpm
|
| D | Dockerfile.rbe.cuda10.1-cudnn7-centos6 | 7 # 1. Download tensorrt from
8 # https://developer.nvidia.com/nvidia-tensorrt-5x-download
14 # tensorrt for centos is only availble via a authenticated download link from
26 # TODO(klimek): Once nvidia provides a way to install tensorrt from within
27 # docker, switch to that. Until then, we copy the tensorrt rpm into the docker
29 ADD tensorrt.rpm tensorrt.rpm
30 RUN rpm -ihv tensorrt.rpm && \
31 yum install -y --nogpgcheck tensorrt && \
32 rm -f tensorrt.rpm
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | logger_registry.h | 26 namespace tensorrt {
49 static ::tensorflow::tensorrt::RegisterLogger register_trt_logger##ctr \
51 ::tensorflow::tensorrt::RegisterLogger(name, logger)
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | py_utils.i | 63 tensorflow::tensorrt::GetLinkedTensorRTVersion( in get_linked_tensorrt_version()
71 tensorflow::tensorrt::GetLoadedTensorRTVersion( in get_loaded_tensorrt_version()
77 return tensorflow::tensorrt::IsGoogleTensorRTEnabled(); in is_tensorrt_enabled()
|
| D | trt_allocator.h | 30 namespace tensorrt {
40 namespace tensorrt {
|
| D | trt_allocator.cc | 27 namespace tensorrt { namespace
59 namespace tensorrt { namespace
|
| D | py_utils.h | 20 namespace tensorrt {
|
| D | trt_logger.h | 26 namespace tensorrt {
|
| D | py_utils.cc | 24 namespace tensorrt { namespace
|
| /external/tensorflow/tensorflow/python/compiler/ |
| D | BUILD | 18 ["//tensorflow/python/compiler/tensorrt:trt_convert_windows"],
19 otherwise = ["//tensorflow/python/compiler/tensorrt:init_py"],
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.experimental.tensorrt.-converter.pbtxt | 1 path: "tensorflow.experimental.tensorrt.Converter"
3 is_instance: "<class \'tensorflow.python.compiler.tensorrt.trt_convert.TrtGraphConverterV2\'>"
|
| D | tensorflow.experimental.tensorrt.pbtxt | 1 path: "tensorflow.experimental.tensorrt"
|
| D | tensorflow.experimental.pbtxt | 4 name: "tensorrt"
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/plugin/ |
| D | trt_plugin.h | 28 namespace tensorrt {
87 static ::tensorflow::tensorrt::TrtPluginRegistrar<name> \
|
| D | trt_plugin.cc | 24 namespace tensorrt { namespace
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | __init__.py | 22 from tensorflow.python.compiler.tensorrt import trt_convert as trt
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment.h | 31 namespace tensorrt {
|
| /external/tensorflow/tensorflow/tools/ci_build/release/ubuntu_16/gpu_py35_full/ |
| D | pip_v1.sh | 38 export TENSORRT_INSTALL_PATH=/usr/local/tensorrt
|