online-stream-rknn.cc
1.6 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
// sherpa-onnx/csrc/rknn/online-stream-rknn.cc
//
// Copyright (c) 2025 Xiaomi Corporation
#include "sherpa-onnx/csrc/rknn/online-stream-rknn.h"
#include <utility>
#include <vector>
namespace sherpa_onnx {
class OnlineStreamRknn::Impl {
public:
void SetZipformerEncoderStates(std::vector<std::vector<uint8_t>> states) {
states_ = std::move(states);
}
std::vector<std::vector<uint8_t>> &GetZipformerEncoderStates() {
return states_;
}
void SetZipformerResult(OnlineTransducerDecoderResultRknn r) {
result_ = std::move(r);
}
OnlineTransducerDecoderResultRknn &GetZipformerResult() { return result_; }
private:
std::vector<std::vector<uint8_t>> states_;
OnlineTransducerDecoderResultRknn result_;
};
OnlineStreamRknn::OnlineStreamRknn(
const FeatureExtractorConfig &config /*= {}*/,
ContextGraphPtr context_graph /*= nullptr*/)
: OnlineStream(config, context_graph), impl_(std::make_unique<Impl>()) {}
OnlineStreamRknn::~OnlineStreamRknn() = default;
void OnlineStreamRknn::SetZipformerEncoderStates(
std::vector<std::vector<uint8_t>> states) const {
impl_->SetZipformerEncoderStates(std::move(states));
}
std::vector<std::vector<uint8_t>> &OnlineStreamRknn::GetZipformerEncoderStates()
const {
return impl_->GetZipformerEncoderStates();
}
void OnlineStreamRknn::SetZipformerResult(
OnlineTransducerDecoderResultRknn r) const {
impl_->SetZipformerResult(std::move(r));
}
OnlineTransducerDecoderResultRknn &OnlineStreamRknn::GetZipformerResult()
const {
return impl_->GetZipformerResult();
}
} // namespace sherpa_onnx