|
TinyChatEngine
|
Public Member Functions | |
| Fp32GPTBigCodeDecoder (std::string param_path, const struct model_config config) | |
| Matrix3D< float > | prepare_decoder_attention_mask (int length, int past_length) |
| Matrix3D< float > | get_position_embed (int sql_length, int past_length) |
| struct Fp32GPTBigCodeDecoder_output | forward (const struct Fp32GPTBigCodeDecoder_input &input) |
Public Attributes | |
| Embedding | wte |
| Embedding | wpe |
| int | voc_size |
| int | embed_dim |
| int | padding_idx |
| int | hidden_dim |
| int | num_heads |
| int | max_position_embeddings |
| std::vector< Fp32GPTBigCodeDecoderLayer > | layers |
| LayerNorm | ln_f |
| std::string | profile_name = "Fp32GPTBigCodeDecoder" |