diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-04-30 16:43:14 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-30 16:45:53 -0700 |
commit | 1ff23a314f355a9ebaaf207dbeae56ebc1634d63 (patch) | |
tree | 73aead08c95b4443a21597d5ef7c7d21fff0046d /tensorflow/contrib/lite/interpreter.cc | |
parent | 7141ed55dd0f36f698143812b44aeffc6129257b (diff) |
Small fix to prevent a crash if the delegate has not implemented FreeBufferHandle.
PiperOrigin-RevId: 194866595
Diffstat (limited to 'tensorflow/contrib/lite/interpreter.cc')
-rw-r--r-- | tensorflow/contrib/lite/interpreter.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/interpreter.cc b/tensorflow/contrib/lite/interpreter.cc index 9d8ea55fd1..ebb0aedc20 100644 --- a/tensorflow/contrib/lite/interpreter.cc +++ b/tensorflow/contrib/lite/interpreter.cc @@ -125,7 +125,8 @@ Interpreter::~Interpreter() { for (int i = 0; i < context_.tensors_size; i++) { TfLiteTensor* tensor = &context_.tensors[i]; - if (tensor->buffer_handle != kTfLiteNullBufferHandle) { + if (tensor->buffer_handle != kTfLiteNullBufferHandle && + tensor->delegate->FreeBufferHandle != nullptr) { tensor->delegate->FreeBufferHandle(tensor->delegate, &tensor->buffer_handle); } |