speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add functions

Browse Source
This commit is contained in:
lyblsgo 2023-05-10 16:56:12 +08:00
parent 0bddc26393
commit 659e18be11
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
funasr/runtime/onnxruntime/src/util.cpp
View File

@ -180,4 +180,13 @@ void Glu(Tensor<float> *din, Tensor<float> *dout)
}
}
bool is_target_file(const std::string& filename, const std::string target) {
std::size_t pos = filename.find_last_of(".");
if (pos == std::string::npos) {
return false;
}
std::string extension = filename.substr(pos + 1);
return (extension == target);
}
} // namespace funasr

1
funasr/runtime/onnxruntime/src/util.h
View File

@ -25,6 +25,7 @@ extern void FindMax(float *din, int len, float &max_val, int &max_idx);
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