commit ad2d40bcb7c94a9f7d3f710181452bdabc14c954
author Alex Van Damme <atv@google.com> Tue May 07 16:28:33 2024 +0000
committer Gerrit Code Review <noreply-gerritcodereview@google.com> Tue May 07 16:28:33 2024 +0000
tree bf7b2507123ba544c8917a79d10c84e88d8ad5f6
parent b3d19eefba01e091b6fd4829f22c32f3e3bafedc
parent f50dac75dbf329a70f2d00ea4b6dd7b015dd7ebd

Merge "Delegate kernel selection for int8 conv to our library"

tensorflow/lite/micro/kernels/kelvin/depthwise_conv.cc

diff --git a/tensorflow/lite/micro/kernels/kelvin/depthwise_conv.cc b/tensorflow/lite/micro/kernels/kelvin/depthwise_conv.cc
index 489fa66..6e4a28d 100644
--- a/tensorflow/lite/micro/kernels/kelvin/depthwise_conv.cc
+++ b/tensorflow/lite/micro/kernels/kelvin/depthwise_conv.cc
@@ -88,9 +88,11 @@
           break;
         }
         case kTfLiteInt8: {
-          reference_integer_ops::DepthwiseConvPerChannel(
-              DepthwiseConvParamsQuantized(params, data),
-              data.per_channel_output_multiplier, data.per_channel_output_shift,
+          tflite::DepthwiseParams dw_params =
+              DepthwiseConvParamsQuantized(params, data);
+          kelvin::opt::DepthwiseConv2DKelvin(
+              dw_params, data.per_channel_output_multiplier,
+              data.per_channel_output_shift,
               tflite::micro::GetTensorShape(input),
               tflite::micro::GetTensorData<int8_t>(input),
               tflite::micro::GetTensorShape(filter),
@@ -133,14 +135,13 @@
                 data.per_channel_output_shift,
                 dw_params.quantized_activation_min,
                 dw_params.quantized_activation_max,
-                tflite::micro::GetTensorData<int16_t>(output)
-            );
+                tflite::micro::GetTensorData<int16_t>(output));
             break;
           }
 
           reference_integer_ops::DepthwiseConvPerChannel(
-              dw_params,
-              data.per_channel_output_multiplier, data.per_channel_output_shift,
+              dw_params, data.per_channel_output_multiplier,
+              data.per_channel_output_shift,
               tflite::micro::GetTensorShape(input),
               tflite::micro::GetTensorData<int16_t>(input),
               tflite::micro::GetTensorShape(filter),