| 18 | }; |
| 19 | |
| 20 | class RoPEModule { |
| 21 | public: |
| 22 | explicit RoPEModule(RoPEConfig config); |
| 23 | |
| 24 | core::TensorValue build( |
| 25 | core::ModuleBuildContext & ctx, |
| 26 | const core::TensorValue & input, |
| 27 | const core::TensorValue & positions, |
| 28 | const core::TensorValue * frequency_factors = nullptr) const; |
| 29 | |
| 30 | private: |
| 31 | RoPEConfig config_; |
| 32 | }; |
| 33 | |
| 34 | } // namespace engine::modules |
no outgoing calls
no test coverage detected