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/conv.cc

diff --git a/tensorflow/lite/micro/kernels/kelvin/conv.cc b/tensorflow/lite/micro/kernels/kelvin/conv.cc
index 28ec1e8..2e15dd7 100644
--- a/tensorflow/lite/micro/kernels/kelvin/conv.cc
+++ b/tensorflow/lite/micro/kernels/kelvin/conv.cc
@@ -136,13 +136,7 @@
         }
         case kTfLiteInt8: {
           const auto params_q = ConvParamsQuantized(params, data);
-          bool opt = !(params_q.padding_values.width > 0 ||
-              params_q.padding_values.height > 0 ||
-              params_q.dilation_width_factor > 1 ||
-              params_q.dilation_height_factor > 1);
-          auto fn = kelvin::opt::conv_per_channel_b8;
-          if (!opt) fn = reference_integer_ops::ConvPerChannel;
-          fn(
+          kelvin::opt::conv_per_channel_b8(
               params_q, data.per_channel_output_multiplier,
               data.per_channel_output_shift,
               tflite::micro::GetTensorShape(input),