EngineX-Iluvatar/enginex-mr_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Fix style issues for online punctuation source files (#1225)

Browse Source
This commit is contained in:
Fangjun Kuang
2024-08-06 17:43:24 +08:00
committed by GitHub
parent 1414e4dc61
commit 375c055ff8
10 changed files with 137 additions and 123 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
sherpa-onnx/csrc/online-cnn-bilstm-model.h
View File

@@ -23,12 +23,11 @@ namespace sherpa_onnx {
*/
class OnlineCNNBiLSTMModel {
public:
explicit OnlineCNNBiLSTMModel(
const OnlinePunctuationModelConfig &config);
explicit OnlineCNNBiLSTMModel(const OnlinePunctuationModelConfig &config);
#if __ANDROID_API__ >= 9
OnlineCNNBiLSTMModel(AAssetManager *mgr,
const OnlinePunctuationModelConfig &config);
const OnlinePunctuationModelConfig &config);
#endif
~OnlineCNNBiLSTMModel();
@@ -43,7 +42,9 @@ class OnlineCNNBiLSTMModel {
* - case_logits: A 2-D tensor of shape (T', num_cases).
* - punct_logits: A 2-D tensor of shape (T', num_puncts).
*/
std::pair<Ort::Value, Ort::Value> Forward(Ort::Value token_ids, Ort::Value valid_ids, Ort::Value label_lens) const;
std::pair<Ort::Value, Ort::Value> Forward(Ort::Value token_ids,
Ort::Value valid_ids,
Ort::Value label_lens) const;
/** Return an allocator for allocating memory
*/