Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dpy_utils_wrapper.cc30 PYBIND11_MODULE(_pywrap_py_utils, m) { in PYBIND11_MODULE() argument
/external/tensorflow/tensorflow/python/platform/
Dbuild_info_test.py38 from tensorflow.compiler.tf2tensorrt._pywrap_py_utils import is_tensorrt_enabled
/external/tensorflow/tensorflow/compiler/tf2tensorrt/
DBUILD276 ":_pywrap_py_utils",
614 name = "_pywrap_py_utils",
617 module_name = "_pywrap_py_utils",
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py69 _pywrap_py_utils = LazyLoader( variable
198 linked_version = _pywrap_py_utils.get_linked_tensorrt_version()
199 loaded_version = _pywrap_py_utils.get_loaded_tensorrt_version()
DBUILD45 "//tensorflow/compiler/tf2tensorrt:_pywrap_py_utils",
Dtrt_convert_test.py29 from tensorflow.compiler.tf2tensorrt._pywrap_py_utils import is_tensorrt_enabled
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dquantization_mnist_test.py22 from tensorflow.compiler.tf2tensorrt._pywrap_py_utils import is_tensorrt_enabled
Dtf_trt_integration_test_base.py34 from tensorflow.compiler.tf2tensorrt._pywrap_py_utils import get_linked_tensorrt_version
35 from tensorflow.compiler.tf2tensorrt._pywrap_py_utils import is_tensorrt_enabled