aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/interpreter.cc
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-04-30 16:43:14 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-04-30 16:45:53 -0700
commit1ff23a314f355a9ebaaf207dbeae56ebc1634d63 (patch)
tree73aead08c95b4443a21597d5ef7c7d21fff0046d /tensorflow/contrib/lite/interpreter.cc
parent7141ed55dd0f36f698143812b44aeffc6129257b (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.cc3
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);
}