diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-02-08 14:16:38 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-02-08 14:30:40 -0800 |
| commit | 06569522503f79fe950cab36ef00b23ddeaf8fe0 (patch) | |
| tree | 85cbfa9dd296adaf863b56246fa6a810e1b6453b /tensorflow/c/checkpoint_reader.cc | |
| parent | 7c27bdf8b986ea3ceda0c6ab555b52605bfb3b0a (diff) | |
Fix memory leakage of CheckpointReader
Change: 146955950
Diffstat (limited to 'tensorflow/c/checkpoint_reader.cc')
| -rw-r--r-- | tensorflow/c/checkpoint_reader.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/c/checkpoint_reader.cc b/tensorflow/c/checkpoint_reader.cc index 17b3f93193..e7b9bca5b5 100644 --- a/tensorflow/c/checkpoint_reader.cc +++ b/tensorflow/c/checkpoint_reader.cc @@ -58,6 +58,7 @@ CheckpointReader::CheckpointReader(const string& filename, CheckpointReader::~CheckpointReader() { delete var_to_shape_map_ptr_; delete reader_; + delete v2_reader_; } bool CheckpointReader::HasTensor(const string& name) const { |