Loading...
Searching...
No Matches
opennn::GroupedQueryAttentionOperator Member List

This is the complete list of members for opennn::GroupedQueryAttentionOperator, including all inherited members.

back_propagate(ForwardPropagation &, BackPropagation &, size_t) const overrideopennn::GroupedQueryAttentionOperatorvirtual
compute_dtypeopennn::Operator
forward_propagate(ForwardPropagation &, size_t, ForwardPropagationMode) overrideopennn::GroupedQueryAttentionOperatorvirtual
forward_scratch_slots_countopennn::GroupedQueryAttentionOperatorstatic
forward_scratch_specs() constopennn::GroupedQueryAttentionOperator
from_JSON(const Json *)opennn::Operatorinlinevirtual
get_input(ForwardPropagation &forward_propagation, size_t layer, size_t slot_index=0) const noexceptopennn::Operatorinline
get_input_delta(BackPropagation &back_propagation, size_t layer, size_t slot_index=0) const noexceptopennn::Operatorinline
get_inputs(ForwardPropagation &forward_propagation, size_t layer) const noexceptopennn::Operatorinline
get_output(ForwardPropagation &forward_propagation, size_t layer, size_t slot_index=0) const noexceptopennn::Operatorinline
get_output_delta(BackPropagation &back_propagation, size_t layer, size_t slot_index=0) const noexceptopennn::Operatorinline
get_rope_tables() const noexceptopennn::GroupedQueryAttentionOperatorinline
GroupedQueryAttention classopennn::GroupedQueryAttentionOperatorfriend
GroupedQueryAttentionOperator()opennn::GroupedQueryAttentionOperator
initialize_states()opennn::Operatorinlinevirtual
input_delta_slotsopennn::Operator
input_slotsopennn::Operator
link_gradients(span< const TensorView > views)opennn::Operatorinlinevirtual
link_parameter_scales(span< const TensorView >) overrideopennn::GroupedQueryAttentionOperatorvirtual
link_parameters(span< const TensorView >) overrideopennn::GroupedQueryAttentionOperatorvirtual
link_slots(span< const TensorView > views, TensorView *ParameterSlot::*member)opennn::Operatorinline
link_states(span< const TensorView >)opennn::Operatorinlinevirtual
load_state_from_JSON(const Json *)opennn::Operatorinlinevirtual
output_delta_slotsopennn::Operator
output_slotsopennn::Operator
parameter_quantization() const overrideopennn::GroupedQueryAttentionOperatorvirtual
parameter_slots()opennn::Operatorinlinevirtual
parameter_specs() const overrideopennn::GroupedQueryAttentionOperatorvirtual
set(Index new_sequence_length, Index new_hidden, Index new_q_heads, Index new_kv_heads, Index new_head_dim, float new_rope_theta, float new_rms_epsilon, bool new_use_qk_norm)opennn::GroupedQueryAttentionOperator
set_parameters_glorot()opennn::Operatorinlinevirtual
set_parameters_pytorch()opennn::Operatorinlinevirtual
set_parameters_random() overrideopennn::GroupedQueryAttentionOperatorvirtual
set_weights_dtype(Type new_weights_dtype)opennn::Operatorinlinevirtual
state_specs() constopennn::Operatorinlinevirtual
to_JSON(JsonWriter &) constopennn::Operatorinlinevirtual
weights_dtypeopennn::Operator
~GroupedQueryAttentionOperator() overrideopennn::GroupedQueryAttentionOperator
~Operator()=defaultopennn::Operatorvirtual