| /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/tensorrt/ |
| D | tensorrt_configure.bzl | 55 return repository_ctx.path(Label("//third_party/tensorrt:%s.tpl" % filename))
72 _tpl(repository_ctx, "tensorrt/include/tensorrt_config.h", {
79 Label("//third_party/tensorrt:LICENSE"),
87 "tensorrt/tensorrt_config.py",
108 …"tensorrt/include/tensorrt_config.h": _tpl_path(repository_ctx, "tensorrt/include/tensorrt_config.…
109 "tensorrt/tensorrt_config.py": _tpl_path(repository_ctx, "tensorrt/tensorrt_config.py"),
112 config = find_cuda_config(repository_ctx, find_cuda_config_path, ["tensorrt"])
127 outs = ["tensorrt/lib/" + library for library in libraries],
133 outs = ["tensorrt/include/" + header for header in headers],
146 outs = ["tensorrt/lib/" + library for library in static_libraries],
[all …]
|
| D | BUILD.tpl | 21 "tensorrt/include/tensorrt_config.h",
24 include_prefix = "third_party/tensorrt",
25 strip_include_prefix = "tensorrt/include",
29 name = "tensorrt",
57 srcs = ["tensorrt/tensorrt_config.py"]
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | py_utils_wrapper.cc | 23 return tensorflow::tensorrt::GetLinkedTensorRTVersion(); in get_linked_tensorrt_version()
27 return tensorflow::tensorrt::GetLoadedTensorRTVersion(); in get_loaded_tensorrt_version()
38 m.def("is_tensorrt_enabled", tensorflow::tensorrt::IsGoogleTensorRTEnabled, in PYBIND11_MODULE()
|
| D | trt_shape_optimization_profiles.h | 38 namespace tensorrt {
54 return StrCat("[min: ", tensorflow::tensorrt::DebugString(min), in DebugString()
55 ", opt: : ", tensorflow::tensorrt::DebugString(opt), in DebugString()
56 ", max: ", tensorflow::tensorrt::DebugString(max), "]"); in DebugString()
83 << ::tensorflow::tensorrt::DebugString(opt[i]); in SetDimensions()
|
| D | trt_allocator.h | 28 namespace tensorrt {
37 namespace tensorrt {
|
| D | trt_allocator.cc | 25 namespace tensorrt { namespace
56 namespace tensorrt { namespace
|
| D | py_utils.h | 20 namespace tensorrt {
|
| D | py_utils.cc | 25 namespace tensorrt { namespace
|
| D | trt_logger.h | 25 namespace tensorrt {
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.experimental.tensorrt.-conversion-params.pbtxt | 1 path: "tensorflow.experimental.tensorrt.ConversionParams"
3 is_instance: "<class \'tensorflow.python.compiler.tensorrt.trt_convert.TrtConversionParams\'>"
4 is_instance: "<class \'tensorflow.python.compiler.tensorrt.trt_convert.TrtConversionParams\'>"
|
| 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"
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | logger_registry.h | 27 namespace tensorrt {
50 static ::tensorflow::tensorrt::RegisterLogger register_trt_logger##ctr \
52 ::tensorflow::tensorrt::RegisterLogger(name, logger)
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/ |
| D | BUILD | 26 "//tensorflow/python/compiler/tensorrt:trt_convert_py",
41 "//tensorflow/python/compiler/tensorrt:trt_convert_py",
57 "//tensorflow/python/compiler/tensorrt:trt_convert_py",
|
| D | run_models.py | 26 from tensorflow.python.compiler.tensorrt import trt_convert as trt
27 from tensorflow.python.compiler.tensorrt.model_tests import model_handler
28 from tensorflow.python.compiler.tensorrt.model_tests import result_analyzer
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/common/ |
| D | utils.h | 22 namespace tensorrt {
39 namespace tensorrt {
|
| D | utils.cc | 25 namespace tensorrt { namespace
54 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/build_info/ |
| D | gen_build_info.py | 33 from tensorrt.tensorrt import tensorrt_config # pylint: disable=g-import-not-at-top
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/plugin/ |
| D | trt_plugin.h | 27 namespace tensorrt {
86 static ::tensorflow::tensorrt::TrtPluginRegistrar<name> \
|
| D | trt_plugin.cc | 23 namespace tensorrt { namespace
|
| /external/tensorflow/tensorflow/tools/ci_build/ |
| D | pylint_allowlist | 16 ^tensorflow/python/compiler/tensorrt/test/base_test.py.*\[E1003.*bad-super-call
17 ^tensorflow/python/compiler/tensorrt/test/biasadd_matmul_test.py.*\[E1003.*bad-super-call
18 ^tensorflow/python/compiler/tensorrt/test/binary_tensor_weight_broadcast_test.py.*\[E1003.*bad-supe…
19 ^tensorflow/python/compiler/tensorrt/test/dynamic_input_shapes_test.py.*\[E1003.*bad-super-call
20 ^tensorflow/python/compiler/tensorrt/test/int32_test.py.*\[E1003.*bad-super-call
21 ^tensorflow/python/compiler/tensorrt/test/testdata/gen_tftrt_model.py.*\[E1102.*not-callable
22 ^tensorflow/python/compiler/tensorrt/test/vgg_block_nchw_test.py.*\[E1003.*bad-super-call
23 ^tensorflow/python/compiler/tensorrt/test/vgg_block_test.py.*\[E1003.*bad-super-call
|
| /external/tensorflow/tensorflow/ |
| D | opensource_only.files | 215 tensorflow/third_party/tensorrt/BUILD
216 tensorflow/third_party/tensorrt/BUILD.tpl
217 tensorflow/third_party/tensorrt/LICENSE
218 tensorflow/third_party/tensorrt/build_defs.bzl.tpl
219 tensorflow/third_party/tensorrt/tensorrt/include/tensorrt_config.h.tpl
220 tensorflow/third_party/tensorrt/tensorrt/tensorrt_config.py.tpl
221 tensorflow/third_party/tensorrt/tensorrt_configure.bzl
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | __init__.py | 22 from tensorflow.python.compiler.tensorrt import trt_convert as trt
|