Int4GPTBigCodeDecoder Class Reference

Public Member Functions
	Int4GPTBigCodeDecoder (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 Int4GPTBigCodeDecoder_output	forward (const struct Int4GPTBigCodeDecoder_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< Int4GPTBigCodeDecoderLayer >	layers
LayerNorm	ln_f
std::string	profile_name = "Int4GPTBigCodeDecoder"

---

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

• llm/include/nn_modules/Int4GPTBigCodeDecoder.h
• llm/src/nn_modules/Int4GPTBigCodeDecoder.cc