|
OpenNN
Open-source neural networks library
|
This is the complete list of members for opennn::AttentionOp, 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::AttentionOp | |
| 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_delta, TensorView &attention_weight_delta, TensorView &query_delta, TensorView &key_delta, TensorView &value_delta) const | opennn::AttentionOp | |
| attention_output_slots | opennn::AttentionOp | |
| AttentionOp() | opennn::AttentionOp | |
| AttentionOp(AttentionOp &&) noexcept | opennn::AttentionOp | |
| AttentionOp(const AttentionOp &)=delete | opennn::AttentionOp | |
| back_propagate(ForwardPropagation &fp, BackPropagation &bp, size_t layer) const noexcept override | opennn::AttentionOp | virtual |
| causal_mask | opennn::AttentionOp | |
| compute_dtype | opennn::AttentionOp | |
| destroy_cuda() override | opennn::AttentionOp | virtual |
| dropout | opennn::AttentionOp | |
| forward_propagate(ForwardPropagation &fp, size_t layer, bool is_training) noexcept override | opennn::AttentionOp | virtual |
| forward_scratch_specs(Index batch_size) const | opennn::AttentionOp | |
| from_JSON(const Json *parent) override | opennn::AttentionOp | virtual |
| get_input(ForwardPropagation &fp, size_t layer, size_t i=0) const noexcept | opennn::Operator | inline |
| get_input_delta(BackPropagation &bp, size_t layer, size_t i=0) const noexcept | opennn::Operator | inline |
| get_inputs(ForwardPropagation &fp, size_t layer, size_t i=0) const noexcept | opennn::Operator | inline |
| get_output(ForwardPropagation &fp, size_t layer, size_t i=0) const noexcept | opennn::Operator | inline |
| get_output_delta(BackPropagation &bp, size_t layer, size_t i=0) const noexcept | opennn::Operator | inline |
| head_dimension | opennn::AttentionOp | |
| heads_number | opennn::AttentionOp | |
| input_delta_slots | opennn::Operator | |
| input_slots | opennn::Operator | |
| link_gradients(span< const TensorView >) | opennn::Operator | inlinevirtual |
| link_parameters(span< const TensorView >) | opennn::Operator | inlinevirtual |
| link_states(span< const TensorView >) | opennn::Operator | inlinevirtual |
| load_state_from_JSON(const Json *) | opennn::Operator | inlinevirtual |
| operator=(AttentionOp &&) noexcept | opennn::AttentionOp | |
| operator=(const AttentionOp &)=delete | opennn::AttentionOp | |
| output_delta_slots | opennn::Operator | |
| output_slots | opennn::Operator | |
| parameter_specs() const | opennn::Operator | inlinevirtual |
| query_sequence_length | opennn::AttentionOp | |
| scratch_slots | opennn::AttentionOp | |
| set(Index heads_number, Index head_dimension, Index query_sequence_length, Index source_sequence_length, bool use_causal_mask, Type compute_dtype) | opennn::AttentionOp | |
| set_dropout_rate(float rate) | opennn::AttentionOp | inline |
| set_parameters_glorot() | opennn::Operator | inlinevirtual |
| set_parameters_random() | opennn::Operator | inlinevirtual |
| source_sequence_length | opennn::AttentionOp | |
| source_view_index | opennn::AttentionOp | |
| state_specs() const | opennn::Operator | inlinevirtual |
| to_JSON(JsonWriter &w) const override | opennn::AttentionOp | virtual |
| use_causal_mask | opennn::AttentionOp | |
| ~AttentionOp() override | opennn::AttentionOp | |
| ~Operator()=default | opennn::Operator | virtual |