AttentionMask() const | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | inline |
AttentionMask() | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | inline |
Backward(const arma::Mat< eT > &, const arma::Mat< eT > &gy, arma::Mat< eT > &g) | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | |
Delta() const | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | inline |
Delta() | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | inline |
EmbedDim() const | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | inline |
EmbedDim() | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | inline |
Forward(const arma::Mat< eT > &input, arma::Mat< eT > &output) | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | |
Gradient(const arma::Mat< eT > &input, const arma::Mat< eT > &error, arma::Mat< eT > &gradient) | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | |
Gradient() const | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | inline |
Gradient() | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | inline |
InputShape() const | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | inline |
KeyPaddingMask() const | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | inline |
KeyPaddingMask() | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | inline |
MultiheadAttention() | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | |
MultiheadAttention(const size_t tgtSeqLen, const size_t srcSeqLen, const size_t embedDim, const size_t numHeads) | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | |
NumHeads() const | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | inline |
NumHeads() | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | inline |
OutputParameter() const | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | inline |
OutputParameter() | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | inline |
Parameters() const | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | inline |
Parameters() | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | inline |
Reset() | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | |
serialize(Archive &ar, const uint32_t) | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | |
SrcSeqLen() const | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | inline |
SrcSeqLen() | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | inline |
TgtSeqLen() const | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | inline |
TgtSeqLen() | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | inline |
WeightSize() const | MultiheadAttention< InputDataType, OutputDataType, RegularizerType > | inline |