OpenNN
Open-source neural networks library
Loading...
Searching...
No Matches
opennn::MultiHeadProjection Member List

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

apply(const TensorView &input, TensorView &head_output, float *scratch)opennn::MultiHeadProjection
apply_delta(const TensorView &head_gradient, const TensorView &input, TensorView &input_gradient, bool accumulate, float *scratch) constopennn::MultiHeadProjection
combinationopennn::MultiHeadProjection
compute_dtypeopennn::MultiHeadProjection
destroy_cuda()opennn::Operatorinlinevirtual
forward_propagate(ForwardPropagation &, size_t, bool) noexceptopennn::Operatorinlinevirtual
from_JSON(const Json *)opennn::Operatorinlinevirtual
head_dimensionopennn::MultiHeadProjection
heads_numberopennn::MultiHeadProjection
input_featuresopennn::MultiHeadProjection
input_slotsopennn::Operator
link_gradients(const vector< TensorView > &views) overrideopennn::MultiHeadProjectioninlinevirtual
link_parameters(const vector< TensorView > &views) overrideopennn::MultiHeadProjectioninlinevirtual
link_states(const vector< TensorView > &)opennn::Operatorinlinevirtual
load_state_from_JSON(const Json *)opennn::Operatorinlinevirtual
output_slotsopennn::Operator
parameter_specs() const overrideopennn::MultiHeadProjectioninlinevirtual
set(Index input_features, Index heads_number, Index head_dimension, Type compute_dtype)opennn::MultiHeadProjection
set_parameters_glorot() overrideopennn::MultiHeadProjectioninlinevirtual
set_parameters_random() overrideopennn::MultiHeadProjectioninlinevirtual
state_specs() constopennn::Operatorinlinevirtual
to_JSON(JsonWriter &) constopennn::Operatorinlinevirtual
~Operator()=defaultopennn::Operatorvirtual