← Back to C-Kernel-Engine Docs Doxygen Source Documentation
TransformerModel Struct Reference

#include <ckernel_model.h>

Data Fields

CKModelConfig cfg
 
size_t elem_bytes
 
size_t embedded_input_offset
 
size_t final_ln_bias_offset
 
size_t final_ln_weight_offset
 
size_t final_output_offset
 
CKLayerLayoutlayers
 
size_t layers_start_offset
 
size_t lm_head_weight_offset
 
size_t logits_offset
 
uint8_t * memory_base
 
size_t pos_emb_offset
 
size_t token_emb_offset
 
size_t total_bytes
 
size_t total_floats
 

Detailed Description

Definition at line 38 of file ckernel_model.h.

Field Documentation

◆ cfg

CKModelConfig TransformerModel::cfg

Definition at line 39 of file ckernel_model.h.

Referenced by ck_model_load_weights_flat(), and layout_transformer_from_ir().

◆ elem_bytes

size_t TransformerModel::elem_bytes

Definition at line 45 of file ckernel_model.h.

Referenced by layout_transformer_from_ir().

◆ embedded_input_offset

size_t TransformerModel::embedded_input_offset

Definition at line 50 of file ckernel_model.h.

Referenced by layout_transformer_from_ir().

◆ final_ln_bias_offset

size_t TransformerModel::final_ln_bias_offset

Definition at line 54 of file ckernel_model.h.

Referenced by ck_model_load_weights_flat(), and layout_transformer_from_ir().

◆ final_ln_weight_offset

size_t TransformerModel::final_ln_weight_offset

Definition at line 53 of file ckernel_model.h.

Referenced by ck_model_load_weights_flat(), and layout_transformer_from_ir().

◆ final_output_offset

size_t TransformerModel::final_output_offset

Definition at line 55 of file ckernel_model.h.

Referenced by layout_transformer_from_ir().

◆ layers

CKLayerLayout* TransformerModel::layers

Definition at line 61 of file ckernel_model.h.

Referenced by ck_model_load_weights_flat(), and layout_transformer_from_ir().

◆ layers_start_offset

size_t TransformerModel::layers_start_offset

Definition at line 51 of file ckernel_model.h.

Referenced by layout_transformer_from_ir().

◆ lm_head_weight_offset

size_t TransformerModel::lm_head_weight_offset

Definition at line 57 of file ckernel_model.h.

Referenced by ck_model_load_weights_flat(), and layout_transformer_from_ir().

◆ logits_offset

size_t TransformerModel::logits_offset

Definition at line 58 of file ckernel_model.h.

Referenced by layout_transformer_from_ir().

◆ memory_base

uint8_t* TransformerModel::memory_base

Definition at line 42 of file ckernel_model.h.

Referenced by ck_model_load_weights_flat().

◆ pos_emb_offset

size_t TransformerModel::pos_emb_offset

Definition at line 49 of file ckernel_model.h.

Referenced by ck_model_load_weights_flat(), and layout_transformer_from_ir().

◆ token_emb_offset

size_t TransformerModel::token_emb_offset

Definition at line 48 of file ckernel_model.h.

Referenced by ck_model_load_weights_flat(), and layout_transformer_from_ir().

◆ total_bytes

size_t TransformerModel::total_bytes

Definition at line 43 of file ckernel_model.h.

Referenced by layout_transformer_from_ir().

◆ total_floats

size_t TransformerModel::total_floats

Definition at line 44 of file ckernel_model.h.

Referenced by layout_transformer_from_ir().


The documentation for this struct was generated from the following file: