1
1
mirror of https://github.com/OpenVoiceOS/OpenVoiceOS synced 2024-12-25 16:22:58 +01:00
OpenVoiceOS/buildroot-external/package/whisper-tflite/0002-Use-4-threads.patch
j1nx 876ee82daa Extend the whole TensorFlow-Lite runtime infrastructure
- TFlite-runtime GPU delegate support
- TFlite-runtime external delegate support
- ARM Compute Library and ARMnn Delegate
- Whisper-TFlite updates.
2023-01-25 14:22:22 +01:00

25 lines
730 B
Diff

From e2ebeaca6a16d981b0d130c06cfe38592623fbe3 Mon Sep 17 00:00:00 2001
From: j1nx <p.steenbergen@j1nx.nl>
Date: Fri, 20 Jan 2023 15:44:40 +0100
Subject: [PATCH 1/1] Use 4 threads
---
tflite_minimal/minimal.cc | 1 +
1 file changed, 1 insertion(+)
diff --git a/tflite_minimal/minimal.cc b/tflite_minimal/minimal.cc
index cd045e0..88e21fe 100644
--- a/tflite_minimal/minimal.cc
+++ b/tflite_minimal/minimal.cc
@@ -186,6 +186,7 @@ int main(int argc, char* argv[]) {
tflite::InterpreterBuilder builder(*model, resolver);
std::unique_ptr<tflite::Interpreter> interpreter;
builder(&interpreter);
+ interpreter->SetNumThreads(4);
TFLITE_MINIMAL_CHECK(interpreter != nullptr);
// Allocate tensor buffers.
--
2.34.1