diff options
-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; |