diff --git a/tensorflow/lite/micro/kernels/transpose_conv.cc b/tensorflow/lite/micro/kernels/transpose_conv.cc index 0545b1a0d2e323e29e024c578e36909201ec2e80..dcf007c59f8a59f1f6849c35bc5a1b2e19340ded 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 9e809d4a8507eafb93c7ff0717d380d983113206..ec661e01d92a1303d4911e1a5fb01d89ef8b9eec 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; }