patch/0017-Add-cros-mtk-pre-compile-custom-op.patch

From 678798b2c0a7b17e85470cc4d19d759150bded80 Mon Sep 17 00:00:00 2001
From: Yi Chou <yich@google.com>
Date: Fri, 23 Aug 2024 13:13:34 +0800
Subject: [PATCH] Add cros-mtk-pre-compile custom op

---
 tensorflow/lite/tools/benchmark/benchmark_tflite_model.cc        | 1 +
 .../lite/tools/evaluation/stages/tflite_inference_stage.cc       | 1 +
 2 files changed, 2 insertions(+)

diff --git a/tensorflow/lite/tools/benchmark/benchmark_tflite_model.cc b/tensorflow/lite/tools/benchmark/benchmark_tflite_model.cc
index b229a1b4..9d575e1d 100644
--- a/tensorflow/lite/tools/benchmark/benchmark_tflite_model.cc
+++ b/tensorflow/lite/tools/benchmark/benchmark_tflite_model.cc
@@ -67,6 +67,7 @@ RegisterSelectedOps(::tflite::MutableOpResolver* resolver) {
   static TfLiteRegistration reg = { nullptr, nullptr, nullptr, nullptr };
   resolver->AddCustom("Convolution2DTransposeBias",
                       &reg);
+  resolver->AddCustom("cros-mtk-pre-compile", &reg);
 }
 
 namespace tflite {
diff --git a/tensorflow/lite/tools/evaluation/stages/tflite_inference_stage.cc b/tensorflow/lite/tools/evaluation/stages/tflite_inference_stage.cc
index bd22f5d1..9c5b0fe0 100644
--- a/tensorflow/lite/tools/evaluation/stages/tflite_inference_stage.cc
+++ b/tensorflow/lite/tools/evaluation/stages/tflite_inference_stage.cc
@@ -142,6 +142,7 @@ TfLiteStatus TfliteInferenceStage::Init(
   }
   TfLiteRegistration reg = { nullptr, nullptr, nullptr, nullptr };
   resolver_->AddCustom("Convolution2DTransposeBias", &reg);
+  resolver_->AddCustom("cros-mtk-pre-compile", &reg);
   RegisterSelectedOps(resolver_.get());
   InterpreterBuilder(*model_, *resolver_)(&interpreter_);
   if (!interpreter_) {