| 67 | }; |
| 68 | |
| 69 | class ConformerConvModule { |
| 70 | public: |
| 71 | explicit ConformerConvModule(ConformerConvModuleConfig config); |
| 72 | core::TensorValue build( |
| 73 | core::ModuleBuildContext & ctx, |
| 74 | const core::TensorValue & input, |
| 75 | const ConformerConvModuleWeights & weights, |
| 76 | const std::optional<core::TensorValue> & keep_mask = std::nullopt) const; |
| 77 | |
| 78 | private: |
| 79 | ConformerConvModuleConfig config_; |
| 80 | }; |
| 81 | |
| 82 | class StreamingConformerConvModule { |
| 83 | public: |
no outgoing calls