diff options
| author |  Jie <jiej@nvidia.com> | 2018-07-17 15:07:50 -0700 |
|---|---|---|
| committer | Jie <jiej@nvidia.com> | 2018-07-17 15:07:50 -0700 |
| commit | 6d0c3acc5a728652b77914a38482f9592578e0df (patch) | |
| tree | 3f927ef1ceb1db1744bc898d952bb413a21a905c /tensorflow/contrib/tensorrt/convert | |
| parent | c8f0730484376c3020804fcf1639f55934b70cbe (diff) | |
1. rename class member variables with trailing underscore
2. change unique_ptr allocator type to base class.
Diffstat (limited to 'tensorflow/contrib/tensorrt/convert')
| -rw-r--r-- | tensorflow/contrib/tensorrt/convert/convert_graph.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/tensorrt/convert/convert_graph.cc b/tensorflow/contrib/tensorrt/convert/convert_graph.cc index bcc867efea..97a26f796c 100644 --- a/tensorflow/contrib/tensorrt/convert/convert_graph.cc +++ b/tensorflow/contrib/tensorrt/convert/convert_graph.cc @@ -829,7 +829,7 @@ tensorflow::Status ConvertAfterShapes(ConversionParams& params) { // The allocator is used to build the engine. The build and the built engine // will be destroyed after we get the serialized engine string, so it's fine // to use unique_ptr here. - std::unique_ptr<TRTDeviceAllocator> alloc; + std::unique_ptr<TRTBaseAllocator> alloc; auto device_alloc = GetDeviceAndAllocator(params, engine); int cuda_device_id = 0; if (device_alloc.first >= 0) { |