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);
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);
}

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

@ -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_";