From c532687e2d238f1f18be0a776f216ede476263f4 Mon Sep 17 00:00:00 2001
From: deqiangc <86809673+deqiangc@users.noreply.github.com>
Date: Wed, 30 Mar 2022 20:20:16 -0700
Subject: [PATCH] Handle no bias vector in transpose_conv kernel (#1004)

Bias vector can be nullptr in tranpose_conv; hence
should do a check before deallocate.

BUG=#993

Co-authored-by: Ting Yan <94130036+tingyan19@users.noreply.github.com>
---
 tensorflow/lite/micro/kernels/transpose_conv.cc        | 4 +++-
 tensorflow/lite/micro/kernels/xtensa/transpose_conv.cc | 4 +++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/tensorflow/lite/micro/kernels/transpose_conv.cc b/tensorflow/lite/micro/kernels/transpose_conv.cc
index 0545b1a0..dcf007c5 100644
--- a/tensorflow/lite/micro/kernels/transpose_conv.cc
+++ b/tensorflow/lite/micro/kernels/transpose_conv.cc
@@ -133,7 +133,9 @@ TfLiteStatus CalculateOpData(TfLiteContext* context, TfLiteNode* node,
     micro_context->DeallocateTempTfLiteTensor(input);
     micro_context->DeallocateTempTfLiteTensor(filter);
     micro_context->DeallocateTempTfLiteTensor(output);
-    micro_context->DeallocateTempTfLiteTensor(bias);
+    if (bias != nullptr) {
+      micro_context->DeallocateTempTfLiteTensor(bias);
+    }
   }
   return kTfLiteOk;
 }
diff --git a/tensorflow/lite/micro/kernels/xtensa/transpose_conv.cc b/tensorflow/lite/micro/kernels/xtensa/transpose_conv.cc
index 9e809d4a..ec661e01 100644
--- a/tensorflow/lite/micro/kernels/xtensa/transpose_conv.cc
+++ b/tensorflow/lite/micro/kernels/xtensa/transpose_conv.cc
@@ -132,7 +132,9 @@ TfLiteStatus CalculateOpData(TfLiteContext* context, TfLiteNode* node,
     micro_context->DeallocateTempTfLiteTensor(input);
     micro_context->DeallocateTempTfLiteTensor(output);
     micro_context->DeallocateTempTfLiteTensor(filter);
-    micro_context->DeallocateTempTfLiteTensor(bias);
+    if (bias != nullptr) {
+      micro_context->DeallocateTempTfLiteTensor(bias);
+    }
   }
   return kTfLiteOk;
 }
-- 
GitLab