This is the complete list of members for pulsatrix::MultiHeadAttentionModule, including all inherited members.

backward(const Tensor &grad_output) override	pulsatrix::MultiHeadAttentionModule	virtual
compute_device() const override	pulsatrix::MultiHeadAttentionModule	inlinevirtual
d_model() const	pulsatrix::MultiHeadAttentionModule	inline
forward(const Tensor &input)	pulsatrix::Module	inline
forward_impl(const Tensor &input) override	pulsatrix::MultiHeadAttentionModule	protectedvirtual
forward_traced(const Tensor &input, NodeId input_node, ComputationGraph &graph, Autograd &autograd)	pulsatrix::Module	inline
head_dim() const	pulsatrix::MultiHeadAttentionModule	inline
is_training() const	pulsatrix::Module	inline
k_norm()	pulsatrix::MultiHeadAttentionModule	inline
k_proj()	pulsatrix::MultiHeadAttentionModule	inline
last_attention_weights() const	pulsatrix::MultiHeadAttentionModule	inline
MultiHeadAttentionModule(int64_t d_model, int64_t num_heads, DeviceBackend *backend, bool use_rope=true, bool use_qk_norm=false)	pulsatrix::MultiHeadAttentionModule
named_parameters() override	pulsatrix::MultiHeadAttentionModule	virtual
num_heads() const	pulsatrix::MultiHeadAttentionModule	inline
op_type() const override	pulsatrix::MultiHeadAttentionModule	inlinevirtual
out_proj()	pulsatrix::MultiHeadAttentionModule	inline
parameters()	pulsatrix::Module	inlinevirtual
propagate_relevance(const Tensor &relevance_out, const LRPRuleConfig &config) override	pulsatrix::MultiHeadAttentionModule	virtual
q_norm()	pulsatrix::MultiHeadAttentionModule	inline
q_proj()	pulsatrix::MultiHeadAttentionModule	inline
set_requires_grad(bool requires_grad, const std::string &prefix="")	pulsatrix::Module	inline
set_training(bool training) override	pulsatrix::MultiHeadAttentionModule	virtual
supports_lrp_rule(LRPRule rule) const	pulsatrix::Module	inlinevirtual
uses_qk_norm() const	pulsatrix::MultiHeadAttentionModule	inline
uses_rope() const	pulsatrix::MultiHeadAttentionModule	inline
v_proj()	pulsatrix::MultiHeadAttentionModule	inline
~Module()=default	pulsatrix::Module	virtual