From 03d4ce829814b4a7f57235fda049351c524ba32b Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期五, 17 三月 2023 14:06:56 +0800
Subject: [PATCH] Merge branch 'main' into dev_xw
---
funasr/runtime/onnxruntime/src/commonfunc.h | 56 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 56 insertions(+), 0 deletions(-)
diff --git a/funasr/runtime/onnxruntime/src/commonfunc.h b/funasr/runtime/onnxruntime/src/commonfunc.h
new file mode 100644
index 0000000..11c234e
--- /dev/null
+++ b/funasr/runtime/onnxruntime/src/commonfunc.h
@@ -0,0 +1,56 @@
+#pragma once
+
+
+typedef struct
+{
+ std::string msg;
+ float snippet_time;
+}RPASR_RECOG_RESULT;
+
+
+#ifdef _WIN32
+#include <codecvt>
+
+
+
+inline std::wstring string2wstring(const std::string& str, const std::string& locale)
+{
+ typedef std::codecvt_byname<wchar_t, char, std::mbstate_t> F;
+ std::wstring_convert<F> strCnv(new F(locale));
+ return strCnv.from_bytes(str);
+}
+
+inline std::wstring strToWstr(std::string str) {
+ if (str.length() == 0)
+ return L"";
+ return string2wstring(str, "zh-CN");
+
+}
+
+#endif
+
+
+
+inline void getInputName(Ort::Session* session, string& inputName,int nIndex=0) {
+ size_t numInputNodes = session->GetInputCount();
+ if (numInputNodes > 0) {
+ Ort::AllocatorWithDefaultOptions allocator;
+ {
+ auto t = session->GetInputNameAllocated(nIndex, allocator);
+ inputName = t.get();
+
+ }
+ }
+}
+
+inline void getOutputName(Ort::Session* session, string& outputName, int nIndex = 0) {
+ size_t numOutputNodes = session->GetOutputCount();
+ if (numOutputNodes > 0) {
+ Ort::AllocatorWithDefaultOptions allocator;
+ {
+ auto t = session->GetOutputNameAllocated(nIndex, allocator);
+ outputName = t.get();
+
+ }
+ }
+}
\ No newline at end of file
--
Gitblit v1.9.1