TinyChatEngine
|
Public Member Functions | |
RotaryPosEmb (Matrix3D< float > _cos, Matrix3D< float > _sin, std::string path) | |
void | forward (Matrix3D< float > &key, Matrix3D< float > &value, int start_idx, int len) |
Public Attributes | |
Matrix3D< float > | cos |
Matrix3D< float > | sin |