commit 1eb552ec21ece648e8b009e137c0657ea556f036
author Naveen Dodda <ndodda@google.com> Mon Apr 08 20:08:56 2024 +0000
committer Naveen Dodda <ndodda@google.com> Tue Apr 09 05:15:17 2024 +0000
tree 195b41a53ee453ca76ed77701002283f28c6905b
parent ff784465c72cd45f4349333a3d3828bcd8201705

Updating resolver to support mobilenet models

Change-Id: Ie030883481daca46cdaa76e2bbb83df0c718255d

benchmarks/benchmark_kelvin.cc

diff --git a/benchmarks/benchmark_kelvin.cc b/benchmarks/benchmark_kelvin.cc
index 6b9e22e..b01ad61 100644
--- a/benchmarks/benchmark_kelvin.cc
+++ b/benchmarks/benchmark_kelvin.cc
@@ -46,7 +46,7 @@
 #endif
 
 namespace {
-constexpr int kTensorArenaSize = 1024 * 1024;
+constexpr int kTensorArenaSize = 1232 * 1024;
 uint8_t g_tensor_arena[kTensorArenaSize] __attribute__((aligned(64)));
 
 __attribute__((section(".model_output_header"))) BenchmarkOutputHeader output_header = {
@@ -56,7 +56,7 @@
 };
 
 // This includes all ops currently used in the Kelvin model suite. More can be added.
-constexpr int kAllOpsNum = 22;
+constexpr int kAllOpsNum = 25;
 std::unique_ptr<tflite::MicroMutableOpResolver<kAllOpsNum>> GetAllOpsResolver() {
   tflite::MicroMutableOpResolver<kAllOpsNum> resolver;
   resolver.AddAveragePool2D();
@@ -81,6 +81,9 @@
   resolver.AddTransposeConv();
   resolver.AddAdd();
   resolver.AddSub();
+  resolver.AddMean();
+  resolver.AddPack();
+  resolver.AddShape();
   return std::make_unique<tflite::MicroMutableOpResolver<kAllOpsNum>>(resolver);
 }