| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | BUILD | 2 # Wrap NVIDIA TensorRT (http://developer.nvidia.com/tensorrt) with tensorflow
33 visibility = ["//tensorflow/python/compiler/tensorrt:__pkg__"],
50 visibility = ["//tensorflow/python/compiler/tensorrt:__pkg__"],
56 visibility = ["//tensorflow/python/compiler/tensorrt:__pkg__"],
66 visibility = ["//tensorflow/python/compiler/tensorrt:__pkg__"],
85 "//tensorflow/python/compiler/tensorrt:tf_trt_integration_test_base",
105 "//tensorflow/python/compiler/tensorrt:tf_trt_integration_test_base",
125 "//tensorflow/python/compiler/tensorrt:tf_trt_integration_test_base",
145 "//tensorflow/python/compiler/tensorrt:tf_trt_integration_test_base",
166 "//tensorflow/python/compiler/tensorrt:tf_trt_integration_test_base",
[all …]
|
| D | tf_function_test.py | 22 from tensorflow.python.compiler.tensorrt import trt_convert
23 from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
24 from tensorflow.python.compiler.tensorrt.test.tf_trt_integration_test_base import GraphState
25 from tensorflow.python.compiler.tensorrt.test.tf_trt_integration_test_base import IsQuantizationWit…
|
| /external/tensorflow/third_party/tensorrt/ |
| D | tensorrt_configure.bzl | 83 return repository_ctx.path(Label("//third_party/tensorrt:%s.tpl" % filename))
101 _tpl(repository_ctx, "tensorrt/include/tensorrt_config.h", {
108 Label("//third_party/tensorrt:LICENSE"),
116 "tensorrt/tensorrt_config.py",
137 …"tensorrt/include/tensorrt_config.h": _tpl_path(repository_ctx, "tensorrt/include/tensorrt_config.…
138 "tensorrt/tensorrt_config.py": _tpl_path(repository_ctx, "tensorrt/tensorrt_config.py"),
142 config = find_cuda_config(repository_ctx, find_cuda_config_path, ["tensorrt"])
157 outs = ["tensorrt/lib/" + library for library in libraries],
163 outs = ["tensorrt/include/" + header for header in headers],
182 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"]
|
| D | workspace.bzl | 19 build_file = "//third_party/tensorrt/plugin:BUILD",
20 patch_file = ["//third_party/tensorrt/plugin:tensorrt_oss.patch"],
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | py_utils_wrapper.cc | 26 return tensorflow::tensorrt::GetLinkedTensorRTVersion(); in get_linked_tensorrt_version()
30 return tensorflow::tensorrt::GetLoadedTensorRTVersion(); in get_loaded_tensorrt_version()
41 m.def("is_tensorrt_enabled", tensorflow::tensorrt::IsGoogleTensorRTEnabled, in PYBIND11_MODULE()
44 tensorflow::tensorrt::GetRegisteredOpConverters, in PYBIND11_MODULE()
|
| D | trt_shape_optimization_profiles.h | 41 namespace tensorrt {
67 return StrCat("[min: ", tensorflow::tensorrt::DebugString(min), in DebugString()
68 ", opt: : ", tensorflow::tensorrt::DebugString(opt), in DebugString()
69 ", max: ", tensorflow::tensorrt::DebugString(max), "]"); in DebugString()
113 << ::tensorflow::tensorrt::DebugString(opt[idx]); in SetDimensions()
122 << ::tensorflow::tensorrt::DebugString(opt[i]); in SetDimensions()
|
| D | py_utils.cc | 28 namespace tensorrt { namespace
52 auto* registry = tensorflow::tensorrt::convert::GetOpConverterRegistry(); in GetRegisteredOpConverters()
|
| D | trt_allocator.h | 29 namespace tensorrt {
38 namespace tensorrt {
|
| D | trt_allocator.cc | 25 namespace tensorrt { namespace
56 namespace tensorrt { namespace
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | BUILD | 2 # Wrap NVIDIA TensorRT (http://developer.nvidia.com/tensorrt) with tensorflow
62 srcs = ["//tensorflow/python/compiler/tensorrt/test:tf_trt_integration_test_base_srcs"],
82 "//tensorflow/python/compiler/tensorrt/test:trt_convert_test_data",
97 "//tensorflow/python/compiler/tensorrt/test:test_utils",
113 srcs = ["//tensorflow/python/compiler/tensorrt/test:quantization_mnist_test_srcs"],
115 "//tensorflow/python/compiler/tensorrt/test:quantization_mnist_test_data",
|
| /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\'>"
|
| /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)
|
| D | op_converter_registry.h | 29 namespace tensorrt {
90 << tensorrt::convert::GetOpConverterRegistry()->Register( \
99 func, tensorrt::convert::kDefaultConverterPriority, __VA_ARGS__)
|
| D | weights.cc | 26 namespace tensorrt { namespace
80 ", type=", tensorflow::tensorrt::DebugString(type_), in DebugString()
180 "tensor=", tensorflow::tensorrt::DebugString(tensor()), in DebugString()
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/ |
| D | BUILD | 26 "//tensorflow/python/compiler/tensorrt:trt_convert_py",
40 "//tensorflow/python/compiler/tensorrt:trt_convert_py",
57 "//tensorflow/python/compiler/tensorrt:trt_convert_py",
|
| D | run_models.py | 25 from tensorflow.python.compiler.tensorrt import trt_convert as trt
26 from tensorflow.python.compiler.tensorrt.model_tests import model_handler
27 from tensorflow.python.compiler.tensorrt.model_tests import result_analyzer
|
| /external/tensorflow/third_party/tensorrt/plugin/ |
| D | BUILD | 20 "@local_config_tensorrt//:tensorrt",
50 include_prefix = "third_party/tensorrt/plugin/efficientNMSPlugin",
56 "@local_config_tensorrt//:tensorrt",
|
| /external/tensorflow/tensorflow/tools/build_info/ |
| D | gen_build_info.py | 26 from tensorrt.tensorrt import tensorrt_config # pylint: disable=g-import-not-at-top
|
| /external/tensorflow/tensorflow/ |
| D | opensource_only.files | 321 third_party/tensorrt/BUILD.tpl:
322 third_party/tensorrt/BUILD:
323 third_party/tensorrt/LICENSE:
324 third_party/tensorrt/build_defs.bzl.tpl:
325 third_party/tensorrt/plugin/BUILD:
326 third_party/tensorrt/tensorrt/include/tensorrt_config.h.tpl:
327 third_party/tensorrt/tensorrt/tensorrt_config.py.tpl:
328 third_party/tensorrt/tensorrt_configure.bzl:
329 third_party/tensorrt/workspace.bzl:
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/plugin/ |
| D | trt_plugin.h | 27 namespace tensorrt {
86 static ::tensorflow::tensorrt::TrtPluginRegistrar<name> \
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/common/ |
| D | utils.h | 26 namespace tensorrt {
88 namespace tensorrt {
|
| D | utils.cc | 29 namespace tensorrt { namespace
58 namespace tensorrt { namespace
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/ |
| D | trt_convert_api_test.cc | 32 namespace tensorrt { namespace
252 StatusOr<GraphDef> result = tensorrt::ConvertAndBuild( in ConvertAndRunFrozenGraph()
272 StatusOr<GraphDef> result = tensorrt::ConvertAndBuild( in ConvertAndRunSavedModel()
|