Home
last modified time | relevance | path

Searched refs:get_loaded_tensorrt_version (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dpy_utils_wrapper.cc26 std::tuple<int, int, int> get_loaded_tensorrt_version() { in get_loaded_tensorrt_version() function
35 m.def("get_loaded_tensorrt_version", get_loaded_tensorrt_version, in PYBIND11_MODULE()
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py199 loaded_version = _pywrap_py_utils.get_loaded_tensorrt_version()