Add RNN LM rescore for offline ASR with modified_beam_search (#125)

2023-04-23 17:15:18 +08:00
parent d49a597431
commit 86017f9833
32 changed files with 842 additions and 52 deletions
--- a/sherpa-onnx/csrc/onnx-utils.h
+++ b/sherpa-onnx/csrc/onnx-utils.h
@@ -86,6 +86,9 @@ std::vector<char> ReadFile(const std::string &filename);
 std::vector<char> ReadFile(AAssetManager *mgr, const std::string &filename);
 #endif

+// TODO(fangjun): Document it
+Ort::Value Repeat(OrtAllocator *allocator, Ort::Value *cur_encoder_out,
+                  const std::vector<int32_t> &hyps_num_split);
 }  // namespace sherpa_onnx

 #endif  // SHERPA_ONNX_CSRC_ONNX_UTILS_H_