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

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

attention_output_slotopennn::AttentionOperator
AttentionOperator()opennn::AttentionOperator
AttentionOperator(AttentionOperator &&) noexceptopennn::AttentionOperator
AttentionOperator(const AttentionOperator &)=deleteopennn::AttentionOperator
back_propagate(ForwardPropagation &, BackPropagation &, size_t) const overrideopennn::AttentionOperatorvirtual
backward_scratch_spec(Index) constopennn::AttentionOperator
causal_maskopennn::AttentionOperator
compute_dtypeopennn::Operator
concatenate_output_heads(ForwardPropagation &, size_t) constopennn::AttentionOperator
concatenated_output_delta_slotopennn::AttentionOperator
dropoutopennn::AttentionOperator
dropout_mask_slotopennn::AttentionOperator
forward_propagate(ForwardPropagation &, size_t, ForwardPropagationMode) overrideopennn::AttentionOperatorvirtual
forward_scratch_specs(Index) constopennn::AttentionOperator
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
head_dimensionopennn::AttentionOperator
heads_numberopennn::AttentionOperator
initialize_states()opennn::Operatorinlinevirtual
input_delta_slotsopennn::Operator
input_slotsopennn::Operator
interleaved_headsopennn::AttentionOperator
link_gradients(span< const TensorView > views)opennn::Operatorinlinevirtual
link_parameter_scales(span< const TensorView >)opennn::Operatorinlinevirtual
link_parameters(span< const TensorView > views)opennn::Operatorinlinevirtual
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
operator=(AttentionOperator &&) noexceptopennn::AttentionOperator
operator=(const AttentionOperator &)=deleteopennn::AttentionOperator
output_delta_slotsopennn::Operator
output_slotsopennn::Operator
parameter_quantization() constopennn::Operatorinlinevirtual
parameter_slots()opennn::Operatorinlinevirtual
parameter_specs() constopennn::Operatorinlinevirtual
query_sequence_lengthopennn::AttentionOperator
scratch_slotopennn::AttentionOperator
sdpa_gradient_scratch_specs(Index) constopennn::AttentionOperator
sdpa_gradient_slotopennn::AttentionOperator
sdpa_qkv_pack_slotopennn::AttentionOperator
sdpa_qkv_pack_spec(Index) constopennn::AttentionOperator
sdpa_scratch_slots_countopennn::AttentionOperatorstatic
sdpa_state_slotopennn::AttentionOperator
sdpa_state_slots_countopennn::AttentionOperatorstatic
sdpa_state_specs(Index) constopennn::AttentionOperator
sdpa_supported(Type, Device)opennn::AttentionOperatorstatic
set(Index, Index, Index, Index, CausalMask, Type)opennn::AttentionOperator
set_parameters_glorot()opennn::Operatorinlinevirtual
set_parameters_pytorch()opennn::Operatorinlinevirtual
set_parameters_random()opennn::Operatorinlinevirtual
set_weights_dtype(Type new_weights_dtype)opennn::Operatorinlinevirtual
source_sequence_lengthopennn::AttentionOperator
split_output_delta(ForwardPropagation &, BackPropagation &, size_t) constopennn::AttentionOperator
state_specs() constopennn::Operatorinlinevirtual
to_JSON(JsonWriter &) constopennn::Operatorinlinevirtual
use_causal_maskopennn::AttentionOperator
use_sdpaopennn::AttentionOperator
weights_dtypeopennn::Operator
zero_padded_queriesopennn::AttentionOperator
~AttentionOperator() overrideopennn::AttentionOperator
~Operator()=defaultopennn::Operatorvirtual