commit d2aca9745d1f504829f2c0630bb6c4f1b672b6bf
author Alex Van Damme <atv@google.com> Wed May 29 13:28:21 2024 -0700
committer Alex Van Damme <atv@google.com> Wed May 29 13:31:27 2024 -0700
tree 9d88a401689229c32d5ee9d3c09ec79666793767
parent 78538b320b26e2650168bf8b1785b368eedef8bb

Add additional ops to benchmark

Change-Id: Ic2188f79093bded3852ea34d8da3a11e15d64f96

benchmarks/benchmark_kelvin.cc

diff --git a/benchmarks/benchmark_kelvin.cc b/benchmarks/benchmark_kelvin.cc
index 7756b34..eac9fe8 100644
--- a/benchmarks/benchmark_kelvin.cc
+++ b/benchmarks/benchmark_kelvin.cc
@@ -52,7 +52,7 @@
 #endif
 
 namespace {
-constexpr int kTensorArenaSize = 1536 * 1024;
+constexpr int kTensorArenaSize = 2 * 1024 * 1024;
 uint8_t g_tensor_arena[kTensorArenaSize] __attribute__((aligned(64)));
 
 __attribute__((section(".model_output_header"))) BenchmarkOutputHeader output_header = {
@@ -63,7 +63,7 @@
 };
 
 // This includes all ops currently used in the Kelvin model suite. More can be added.
-constexpr int kAllOpsNum = 25;
+constexpr int kAllOpsNum = 28;
 std::unique_ptr<tflite::MicroMutableOpResolver<kAllOpsNum>> GetAllOpsResolver() {
   tflite::MicroMutableOpResolver<kAllOpsNum> resolver;
   resolver.AddAveragePool2D();
@@ -91,6 +91,9 @@
   resolver.AddMean();
   resolver.AddPack();
   resolver.AddShape();
+  resolver.AddResizeNearestNeighbor();
+  resolver.AddTranspose();
+  resolver.AddMul();
   return std::make_unique<tflite::MicroMutableOpResolver<kAllOpsNum>>(resolver);
 }