TinyChatEngine
Loading...
Searching...
No Matches
Fp32llamaDecoderLayer Class Reference

Public Member Functions

 Fp32llamaDecoderLayer (std::string param_path, const struct model_config config, int layer_idx)
 
struct Fp32llamaDecoderLayer_output forward (const struct Fp32llamaDecoderLayer_input &input)
 

Public Attributes

int embed_dim
 
int num_attention_heads
 
int hidden_dim
 
int layer_idx
 
float rms_norm_eps
 
LlamaRMSNorm input_layernorm
 
LlamaRMSNorm post_attention_layernorm
 
Linear_FP gate_proj
 
Linear_FP down_proj
 
Linear_FP up_proj
 
Fp32llamaAttention attn
 
std::string profile_name = "Fp32llamaDecoderLayer"
 

The documentation for this class was generated from the following files: