| /external/tensorflow/tensorflow/compiler/tf2tensorrt/ |
| D | BUILD | 2 # Wrap NVIDIA TensorRT (http://developer.nvidia.com/tensorrt) with tensorflow
46 "@local_config_tensorrt//:tensorrt",
72 "@local_config_tensorrt//:tensorrt",
86 "@local_config_tensorrt//:tensorrt",
138 "@local_config_tensorrt//:tensorrt",
155 "@local_config_tensorrt//:tensorrt",
190 "@local_config_tensorrt//:tensorrt",
203 "@local_config_tensorrt//:tensorrt",
274 "@local_config_tensorrt//:tensorrt",
306 "@local_config_tensorrt//:tensorrt",
[all …]
|
| /external/tensorflow/tensorflow/contrib/tensorrt/custom_plugin_examples/ |
| D | BUILD | 2 # Example for plugin support in TensorRT(http://developer.nvidia.com/tensorrt)
51 "@local_config_tensorrt//:tensorrt",
70 "@local_config_tensorrt//:tensorrt",
108 "//tensorflow/contrib/tensorrt:init_py",
|
| D | plugin_test.py | 23 from tensorflow.contrib import tensorrt
24 from tensorflow.contrib.tensorrt import custom_plugin_examples
83 trt_graph = tensorrt.create_inference_graph(
|
| D | __init__.py | 21 from tensorflow.contrib.tensorrt.custom_plugin_examples import inc_op as import_inc_op_so
22 from tensorflow.contrib.tensorrt.custom_plugin_examples.ops import gen_inc_op
|
| D | inc_op_kernel.h | 24 namespace tensorrt {
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_conversion.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 | __init__.py | 22 from tensorflow.python.compiler.tensorrt.trt_convert import create_inference_graph
|
| /external/tensorflow/tensorflow/contrib/tensorrt/ |
| D | BUILD | 2 # Wrap NVIDIA TensorRT (http://developer.nvidia.com/tensorrt) with tensorflow
31 "@local_config_tensorrt//:tensorrt",
44 "//tensorflow/python/compiler/tensorrt:init_py",
|
| D | __init__.py | 22 from tensorflow.contrib.tensorrt.python import *
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| 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 | trt_logger.cc | 23 namespace tensorrt { namespace
|
| D | py_utils.cc | 23 namespace tensorrt { namespace
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | quantization_test.py | 23 from tensorflow.python.compiler.tensorrt.test import tf_trt_integration_test_base as trt_test
24 from tensorflow.python.compiler.tensorrt.wrap_conversion import get_linked_tensorrt_version
|
| D | quantization_mnist_test.py | 24 from tensorflow.python.compiler.tensorrt import trt_convert
25 from tensorflow.python.compiler.tensorrt.wrap_conversion import get_linked_tensorrt_version
26 from tensorflow.python.compiler.tensorrt.wrap_conversion import is_tensorrt_enabled
|
| /external/tensorflow/tensorflow/contrib/tensorrt/python/ |
| D | __init__.py | 22 from tensorflow.contrib.tensorrt.python.trt_convert import create_inference_graph
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/plugin/ |
| D | trt_plugin_utils.h | 29 namespace tensorrt {
|
| D | trt_plugin_utils.cc | 23 namespace tensorrt { namespace
|
| /external/tensorflow/tensorflow/ |
| D | opensource_only.files | 184 tensorflow/third_party/tensorrt/LICENSE
185 tensorflow/third_party/tensorrt/BUILD
186 tensorflow/third_party/tensorrt/build_defs.bzl.tpl
187 tensorflow/third_party/tensorrt/BUILD.tpl
188 tensorflow/third_party/tensorrt/tensorrt_configure.bzl
|
| /external/tensorflow/tensorflow/python/compiler/ |
| D | BUILD | 17 "//tensorflow/python/compiler/tensorrt:init_py",
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | utils.h | 24 namespace tensorrt {
|
| D | utils.cc | 22 namespace tensorrt { namespace
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment.h | 31 namespace tensorrt {
|