experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Address review comments

Browse Source
This commit is contained in:
gracehoney 2018-08-20 16:04:00 -07:00
parent 0b31ebb51b
commit 4684421d9a
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/contrib/tensorrt/convert/convert_graph.cc
View File

@ -794,7 +794,8 @@ std::pair<int, tensorflow::Allocator*> GetDeviceAndAllocator(
gpu_options, tf_gpu_id, 1); gpu_options, tf_gpu_id, 1);
break; 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); return std::make_pair(cuda_device_id, dev_allocator);
} }

1
tensorflow/contrib/tensorrt/convert/convert_nodes.cc
View File

@ -77,6 +77,7 @@ limitations under the License.
namespace tensorflow { namespace tensorflow {
namespace tensorrt { namespace tensorrt {
// TODO(aaroey): put these constants into some class.
const char* const kInputPHName = "TensorRTInputPH_"; const char* const kInputPHName = "TensorRTInputPH_";
const char* const kOutputPHName = "TensorRTOutputPH_"; const char* const kOutputPHName = "TensorRTOutputPH_";