diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-05-08 09:46:45 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-08 14:08:53 -0700 |
commit | 77bb984c23aa7ec347c981c31f650598c9624304 (patch) | |
tree | 137c6ef934938b0c70ddd2c7bf1fc806327f41c2 /tensorflow/contrib/lite/nnapi_delegate.cc | |
parent | bd606508ebb0e1dbb3215c3ad1d0a41da3507766 (diff) |
Free ANeuralNetworksCompilation object in NNAPIDelegate destructor
PiperOrigin-RevId: 195832807
Diffstat (limited to 'tensorflow/contrib/lite/nnapi_delegate.cc')
-rw-r--r-- | tensorflow/contrib/lite/nnapi_delegate.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/nnapi_delegate.cc b/tensorflow/contrib/lite/nnapi_delegate.cc index 6a231dc6bc..eb451397bd 100644 --- a/tensorflow/contrib/lite/nnapi_delegate.cc +++ b/tensorflow/contrib/lite/nnapi_delegate.cc @@ -61,6 +61,10 @@ NNAPIAllocation::~NNAPIAllocation() { } NNAPIDelegate::~NNAPIDelegate() { + if (nn_compiled_model_) { + ANeuralNetworksCompilation_free(nn_compiled_model_); + nn_compiled_model_ = nullptr; + } if (nn_model_) { ANeuralNetworksModel_free(nn_model_); nn_model_ = nullptr; |