Fixes to depthwise_conv for optional bias tensor when compression is …

…enabled.
ddavis-2015 · Dec 13, 2024 · e788456 · e788456
1 parent 4c5079b
commit e788456
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/tensorflow/lite/micro/kernels/depthwise_conv.cc b/tensorflow/lite/micro/kernels/depthwise_conv.cc
@@ -76,8 +76,8 @@ TfLiteStatus DepthwiseConvEval(TfLiteContext* context, TfLiteNode* node) {
                                               filter_comp_td,
                                               data.weights_scratch_index),
           tflite::micro::GetTensorShape(bias),
-          tflite::micro::GetTensorData<float>(micro_context, bias, bias_comp_td,
-                                              data.bias_scratch_index),
+          tflite::micro::GetOptionalTensorData<float>(
+              micro_context, bias, bias_comp_td, data.bias_scratch_index),
 #else   // USE_TFLM_COMPRESSION
           tflite::micro::GetTensorData<float>(filter),
           tflite::micro::GetTensorShape(bias),
@@ -120,7 +120,7 @@ TfLiteStatus DepthwiseConvEval(TfLiteContext* context, TfLiteNode* node) {
                                                    filter_comp_td,
                                                    data.weights_scratch_index),
               tflite::micro::GetTensorShape(bias),
-              tflite::micro::GetTensorData<int32_t>(
+              tflite::micro::GetOptionalTensorData<int32_t>(
                   micro_context, bias, bias_comp_td, data.bias_scratch_index),
 #else   // USE_TFLM_COMPRESSION
               tflite::micro::GetTensorData<int8_t>(filter),
@@ -153,7 +153,7 @@ TfLiteStatus DepthwiseConvEval(TfLiteContext* context, TfLiteNode* node) {
                                                    filter_comp_td,
                                                    data.weights_scratch_index),
               tflite::micro::GetTensorShape(bias),
-              tflite::micro::GetTensorData<int64_t>(
+              tflite::micro::GetOptionalTensorData<int64_t>(
                   micro_context, bias, bias_comp_td, data.bias_scratch_index),
 #else   // USE_TFLM_COMPRESSION
               tflite::micro::GetTensorData<int8_t>(filter),

diff --git a/tensorflow/lite/micro/kernels/xtensa/depthwise_conv.cc b/tensorflow/lite/micro/kernels/xtensa/depthwise_conv.cc
@@ -128,7 +128,7 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
                   micro_context, &filter_int8, filter_comp_td,
                   op_data.reference_op_data.weights_scratch_index),
               tflite::micro::GetTensorShape(bias),
-              tflite::micro::GetTensorData<int32_t>(
+              tflite::micro::GetOptionalTensorData<int32_t>(
                   micro_context, bias, bias_comp_td,
                   op_data.reference_op_data.bias_scratch_index),
 #else   // USE_TFLM_COMPRESSION
@@ -163,7 +163,7 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
                   micro_context, &filter_int8, filter_comp_td,
                   op_data.reference_op_data.weights_scratch_index),
               tflite::micro::GetTensorShape(bias),
-              tflite::micro::GetTensorData<int64_t>(
+              tflite::micro::GetOptionalTensorData<int64_t>(
                   micro_context, bias, bias_comp_td,
                   op_data.reference_op_data.bias_scratch_index),
 #else   // USE_TFLM_COMPRESSION

diff --git a/tensorflow/lite/micro/kernels/xtensa/depthwise_conv_hifi.cc b/tensorflow/lite/micro/kernels/xtensa/depthwise_conv_hifi.cc
@@ -204,7 +204,7 @@ TfLiteStatus DepthwiseConvEvalHifi(TfLiteContext* context, TfLiteNode* node,
           micro_context, filter, filter_comp_td,
           data.reference_op_data.weights_scratch_index),
       tflite::micro::GetTensorShape(bias),
-      tflite::micro::GetTensorData<int32_t>(
+      tflite::micro::GetOptionalTensorData<int32_t>(
           micro_context, bias, bias_comp_td,
           data.reference_op_data.bias_scratch_index),
 #else   // USE_TFLM_COMPRESSION