opennn::Attention Member List

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

apply(const TensorView &query, const TensorView &key, const TensorView &value, const TensorView &source_input, TensorView &attention_weights, TensorView &attention_weights_dropped, TensorView &output, float *mask_scratch, bool is_training)	opennn::Attention
apply_delta(const TensorView &query, const TensorView &key, const TensorView &value, const TensorView &attention_output, const TensorView &attention_weights, const TensorView &attention_weights_dropped, const TensorView &output_gradient, TensorView &attention_weight_gradient, TensorView &query_gradient, TensorView &key_gradient, TensorView &value_gradient) const	opennn::Attention
Attention()	opennn::Attention
Attention(Attention &&) noexcept	opennn::Attention
Attention(const Attention &)=delete	opennn::Attention
causal_mask	opennn::Attention
compute_dtype	opennn::Attention
destroy_cuda() override	opennn::Attention	virtual
dropout	opennn::Attention
forward_propagate(ForwardPropagation &, size_t, bool) noexcept	opennn::Operator	inlinevirtual
forward_scratch_specs(Index batch_size) const	opennn::Attention
from_JSON(const Json *parent) override	opennn::Attention	virtual
head_dimension	opennn::Attention
heads_number	opennn::Attention
input_slots	opennn::Operator
link_gradients(const vector< TensorView > &)	opennn::Operator	inlinevirtual
link_parameters(const vector< TensorView > &)	opennn::Operator	inlinevirtual
link_states(const vector< TensorView > &)	opennn::Operator	inlinevirtual
load_state_from_JSON(const Json *)	opennn::Operator	inlinevirtual
operator=(Attention &&) noexcept	opennn::Attention
operator=(const Attention &)=delete	opennn::Attention
output_slots	opennn::Operator
parameter_specs() const	opennn::Operator	inlinevirtual
query_sequence_length	opennn::Attention
set(Index heads_number, Index head_dimension, Index query_sequence_length, Index source_sequence_length, bool use_causal_mask, Type compute_dtype)	opennn::Attention
set_dropout_rate(float rate)	opennn::Attention	inline
set_parameters_glorot()	opennn::Operator	inlinevirtual
set_parameters_random()	opennn::Operator	inlinevirtual
source_sequence_length	opennn::Attention
state_specs() const	opennn::Operator	inlinevirtual
to_JSON(JsonWriter &w) const override	opennn::Attention	virtual
use_causal_mask	opennn::Attention
~Attention() override	opennn::Attention
~Operator()=default	opennn::Operator	virtual