From cf36ce977c0b8bd54c906e63ce31931ac060178f Mon Sep 17 00:00:00 2001
From: anyvoice <130631963+anyvoiceai@users.noreply.github.com>
Date: 星期日, 02 七月 2023 09:12:18 +0800
Subject: [PATCH] Creates tensor in target device to avoid high CPU occupation. (#695)

---
 funasr/runtime/onnxruntime/src/util.h |    1 +
 1 files changed, 1 insertions(+), 0 deletions(-)

diff --git a/funasr/runtime/onnxruntime/src/util.h b/funasr/runtime/onnxruntime/src/util.h
index 95ef458..8823a32 100644
--- a/funasr/runtime/onnxruntime/src/util.h
+++ b/funasr/runtime/onnxruntime/src/util.h
@@ -25,6 +25,7 @@
 extern void Glu(Tensor<float> *din, Tensor<float> *dout);
 
 string PathAppend(const string &p1, const string &p2);
+bool is_target_file(const std::string& filename, const std::string target);
 
 } // namespace funasr
 #endif

--
Gitblit v1.9.1