fix import

diff --git a/tensorflow/contrib/tensorrt/test/test_tftrt.py b/tensorflow/contrib/tensorrt/test/test_tftrt.py
index 9ba8cbf..58f786c 100644 (file)
--- a/tensorflow/contrib/tensorrt/test/test_tftrt.py
+++ b/tensorflow/contrib/tensorrt/test/test_tftrt.py
@@ -24,7 +24,7 @@ import numpy as np
 # it looks like internal builds don't like it so
 # importing every module individually
 
-from tensorflow.contrib.tensorrt as trt
+from tensorflow.contrib import tensorrt as trt
 from tensorflow.core.protobuf import config_pb2 as cpb2
 from tensorflow.python.client import session as csess
 from tensorflow.python.framework import constant_op as cop