diff options
Diffstat (limited to 'tensorflow/contrib/tensorrt/python/__init__.py')
-rw-r--r-- | tensorflow/contrib/tensorrt/python/__init__.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/tensorrt/python/__init__.py b/tensorflow/contrib/tensorrt/python/__init__.py index 0b2321b5fc..fe4fa166a1 100644 --- a/tensorflow/contrib/tensorrt/python/__init__.py +++ b/tensorflow/contrib/tensorrt/python/__init__.py @@ -22,4 +22,5 @@ from __future__ import print_function from tensorflow.contrib.tensorrt.python.ops import trt_engine_op from tensorflow.contrib.tensorrt.python.trt_convert import calib_graph_to_infer_graph from tensorflow.contrib.tensorrt.python.trt_convert import create_inference_graph +from tensorflow.contrib.tensorrt.python.trt_convert import is_tensorrt_enabled # pylint: enable=unused-import,line-too-long |