Remove changes to tensorboard script
authorSami Kama <skama@nvidia.com>
Wed, 30 May 2018 21:41:31 +0000 (14:41 -0700)
committerSami Kama <skama@nvidia.com>
Wed, 30 May 2018 21:41:31 +0000 (14:41 -0700)
tensorflow/python/tools/import_pb_to_tensorboard.py

index 96f47c8..00de044 100644 (file)
@@ -29,13 +29,6 @@ from tensorflow.python.platform import app
 from tensorflow.python.platform import gfile
 from tensorflow.python.summary import summary
 
-# Try importing TensorRT ops if available
-# pylint: disable=unused-import,trailing-whitespace,g-import-not-at-top,wildcard-import
-try:
-  from tensorflow.contrib.tensorrt.ops.gen_trt_engine_op import *
-except ImportError:
-  pass
-# pylint: enable=unused-import,trailing-whitespace,g-import-not-at-top,wildcard-import
 
 def import_to_tensorboard(model_dir, log_dir):
   """View an imported protobuf model (`.pb` file) as a graph in Tensorboard.