diff options
author | 2018-07-18 22:24:34 -0700 | |
---|---|---|
committer | 2018-07-18 22:24:34 -0700 | |
commit | 3e5dbd6a34e3a069f27e33de341e5f8d4cfdd7b4 (patch) | |
tree | e75390421911d04ee7c647f2f56c7820f5516084 /tensorflow/contrib/tensorrt | |
parent | 5a78e98e877bdca794ffd9e5c4f00da5d2e7ee7d (diff) |
Fix a bug where plugin factory didn't reset the unique_ptr but release it (thus cause mem leak).
Diffstat (limited to 'tensorflow/contrib/tensorrt')
-rw-r--r-- | tensorflow/contrib/tensorrt/plugin/trt_plugin_factory.cc | 3 |
1 files changed, 0 insertions, 3 deletions
diff --git a/tensorflow/contrib/tensorrt/plugin/trt_plugin_factory.cc b/tensorflow/contrib/tensorrt/plugin/trt_plugin_factory.cc index 2bc591484d..cccc912262 100644 --- a/tensorflow/contrib/tensorrt/plugin/trt_plugin_factory.cc +++ b/tensorflow/contrib/tensorrt/plugin/trt_plugin_factory.cc @@ -65,9 +65,6 @@ bool PluginFactoryTensorRT::RegisterPlugin( void PluginFactoryTensorRT::DestroyPlugins() { tensorflow::mutex_lock lock(instance_m_); - for (auto& owned_plugin_ptr : owned_plugins_) { - owned_plugin_ptr.release(); - } owned_plugins_.clear(); } |