Home
last modified time | relevance | path

Searched refs:_pywrap_py_utils (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/
H A Dutils.py23 from tensorflow.compiler.tf2tensorrt import _pywrap_py_utils
70 ver = _pywrap_py_utils.get_linked_tensorrt_version()
75 ver = _pywrap_py_utils.get_loaded_tensorrt_version()
H A Dtrt_convert.py66 _pywrap_py_utils = LazyLoader( variable
113 if (_pywrap_py_utils.is_tensorrt_enabled() and
218 if not _pywrap_py_utils.is_tensorrt_enabled():
230 linked_version = _pywrap_py_utils.get_linked_tensorrt_version()
231 loaded_version = _pywrap_py_utils.get_loaded_tensorrt_version()
H A DBUILD40 "//tensorflow/compiler/tf2tensorrt:_pywrap_py_utils",
H A Dtrt_convert_test.py26 from tensorflow.compiler.tf2tensorrt._pywrap_py_utils import is_tensorrt_enabled
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
H A Dpy_utils_wrapper.cc33 PYBIND11_MODULE(_pywrap_py_utils, m) { in PYBIND11_MODULE() argument
/aosp_15_r20/external/tensorflow/tensorflow/python/platform/
H A Dbuild_info_test.py34 from tensorflow.compiler.tf2tensorrt._pywrap_py_utils import is_tensorrt_enabled
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/
H A DBUILD451 ":_pywrap_py_utils",
999 name = "_pywrap_py_utils",
/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
H A Dquantization_mnist_test.py20 from tensorflow.compiler.tf2tensorrt._pywrap_py_utils import is_tensorrt_enabled
H A Dtf_trt_integration_test_base.py29 from tensorflow.compiler.tf2tensorrt._pywrap_py_utils import is_tensorrt_enabled