Fp32OPTDecoder Class Reference

Public Member Functions
	Fp32OPTDecoder (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 Fp32OPTDecoder_output	forward (const struct Fp32OPTDecoder_input &input)

Public Attributes
Embedding	embed_tokens
Embedding	embed_positions
int	voc_size
int	embed_dim
int	padding_idx
int	hidden_dim
int	num_heads
std::vector< Fp32OPTDecoderLayer >	layers
LayerNorm	final_layer_norm
std::string	profile_name = "Fp32OPTDecoder"

---

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

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