Fp32LlamaForCausalLM_input Struct Reference

Public Member Functions
	Fp32LlamaForCausalLM_input (Matrix3D< int > input_ids_)
	Fp32LlamaForCausalLM_input (Matrix3D< int > input_ids_, std::vector< Matrix3D< float > > past_keys_, std::vector< Matrix3D< float > > past_values_)
	Fp32LlamaForCausalLM_input (Matrix3D< int > input_ids_, Matrix3D< float > image_embed_, Matrix3D< int > second_input_ids_)
	Fp32LlamaForCausalLM_input (Matrix3D< int > input_ids_, Matrix3D< float > image_embed_)

Public Attributes
Matrix3D< int >	input_ids
Matrix3D< float >	image_embed
Matrix3D< int >	second_input_ids
std::vector< Matrix3D< float > >	past_keys
std::vector< Matrix3D< float > >	past_values
bool	has_past_keys_values
bool	is_llava

---

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

• llm/include/nn_modules/Fp32llamaForCausalLM.h