From 2868fe3df4e92a6ae3e327faf6e57ea492e04124 Mon Sep 17 00:00:00 2001
From: 志浩 <neo.dzh@alibaba-inc.com>
Date: 星期四, 16 三月 2023 19:24:21 +0800
Subject: [PATCH] Merge branch 'main' into dev_dzh
---
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