1 files changed, 35 insertions, 0 deletions
diff --git a/tests/ExecuteNetwork/TfliteExecutor.hpp b/tests/ExecuteNetwork/TfliteExecutor.hpp
new file mode 100644
index 0000000000..623d6357eb
--- /dev/null
+++ b/tests/ExecuteNetwork/TfliteExecutor.hpp
@@ -0,0 +1,35 @@
+//
+// Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+#pragma once
+
+#include "IExecutor.hpp"
+#include "NetworkExecutionUtils/NetworkExecutionUtils.hpp"
+#include "ExecuteNetworkProgramOptions.hpp"
+#include "armnn/utility/NumericCast.hpp"
+#include "armnn/utility/Timer.hpp"
+
+#include <armnn_delegate.hpp>
+#include <DelegateOptions.hpp>
+
+#include <tensorflow/lite/c/common.h>
+#include <tensorflow/lite/interpreter.h>
+#include <tensorflow/lite/kernels/register.h>
+
+using namespace tflite;
+class  TfLiteExecutor : public IExecutor
+{
+public:
+    TfLiteExecutor(const ExecuteNetworkParams& m_Params);
+
+    std::vector<const void*> Execute() override;
+    void PrintNetworkInfo() override{};
+    void CompareAndPrintResult(std::vector<const void*> otherOutput) override;
+
+private:
+    std::unique_ptr<tflite::FlatBufferModel> m_Model;
+    const ExecuteNetworkParams& m_Params;
+    std::unique_ptr<Interpreter> m_TfLiteInterpreter;
+};
+