pulsatrix::TransformerBlock Member List

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

backward(const Tensor &grad_output) override	pulsatrix::TransformerBlock	virtual
compute_device() const override	pulsatrix::TransformerBlock	inlinevirtual
d_model() const	pulsatrix::TransformerBlock	inline
forward(const Tensor &input)	pulsatrix::Module	inline
forward_impl(const Tensor &input) override	pulsatrix::TransformerBlock	protectedvirtual
forward_traced(const Tensor &input, NodeId input_node, ComputationGraph &graph, Autograd &autograd)	pulsatrix::Module	inline
is_training() const	pulsatrix::Module	inline
mha()	pulsatrix::TransformerBlock	inline
named_parameters() override	pulsatrix::TransformerBlock	virtual
norm1()	pulsatrix::TransformerBlock	inline
norm2()	pulsatrix::TransformerBlock	inline
op_type() const override	pulsatrix::TransformerBlock	inlinevirtual
parameters()	pulsatrix::Module	inlinevirtual
propagate_relevance(const Tensor &relevance_out, const LRPRuleConfig &config) override	pulsatrix::TransformerBlock	virtual
set_requires_grad(bool requires_grad, const std::string &prefix="")	pulsatrix::Module	inline
set_training(bool training) override	pulsatrix::TransformerBlock	virtual
supports_lrp_rule(LRPRule rule) const	pulsatrix::Module	inlinevirtual
swiglu()	pulsatrix::TransformerBlock	inline
TransformerBlock(int64_t d_model, int64_t num_heads, int64_t d_ff, DeviceBackend *backend, bool use_rope=true, bool use_qk_norm=false)	pulsatrix::TransformerBlock
~Module()=default	pulsatrix::Module	virtual