From 4684421d9aa3e63dc943074025ffdc89df1a1980 Mon Sep 17 00:00:00 2001 From: gracehoney <31743510+aaroey@users.noreply.github.com> Date: Mon, 20 Aug 2018 16:04:00 -0700 Subject: [PATCH] Address review comments --- tensorflow/contrib/tensorrt/convert/convert_graph.cc | 3 ++- tensorflow/contrib/tensorrt/convert/convert_nodes.cc | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/tensorflow/contrib/tensorrt/convert/convert_graph.cc b/tensorflow/contrib/tensorrt/convert/convert_graph.cc index 3dae0ea4e39..b019c99882b 100644 --- a/tensorflow/contrib/tensorrt/convert/convert_graph.cc +++ b/tensorflow/contrib/tensorrt/convert/convert_graph.cc @@ -794,7 +794,8 @@ std::pair GetDeviceAndAllocator( gpu_options, tf_gpu_id, 1); break; } - VLOG(2) << "TF GPU with id " << tf_gpu_id_value << " do not exist " << s; + LOG(ERROR) << "TF GPU with id " << tf_gpu_id_value << " does not exist " + << s; } return std::make_pair(cuda_device_id, dev_allocator); } diff --git a/tensorflow/contrib/tensorrt/convert/convert_nodes.cc b/tensorflow/contrib/tensorrt/convert/convert_nodes.cc index 07b4efd33f3..af52783da11 100644 --- a/tensorflow/contrib/tensorrt/convert/convert_nodes.cc +++ b/tensorflow/contrib/tensorrt/convert/convert_nodes.cc @@ -77,6 +77,7 @@ limitations under the License. namespace tensorflow { namespace tensorrt { +// TODO(aaroey): put these constants into some class. const char* const kInputPHName = "TensorRTInputPH_"; const char* const kOutputPHName = "TensorRTOutputPH_";