TinyChatEngine
|
Public Member Functions | |
Fp32llamaDecoderLayer_input (Matrix3D< float > &hidden_states_, Matrix3D< float > &attention_mask_) | |
Fp32llamaDecoderLayer_input (Matrix3D< float > &hidden_states_, Matrix3D< float > &attention_mask_, Matrix3D< float > past_key_, Matrix3D< float > past_value_) | |
Public Attributes | |
Matrix3D< float > | hidden_states |
Matrix3D< float > | attention_mask |
Matrix3D< float > | past_key |
Matrix3D< float > | past_value |
bool | has_past_key_value = false |