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

backward(const Tensor &grad_output) override	pulsatrix::RWKVModule	virtual
compute_device() const override	pulsatrix::RWKVModule	inlinevirtual
forward(const Tensor &input)	pulsatrix::Module	inline
forward_impl(const Tensor &input) override	pulsatrix::RWKVModule	protectedvirtual
forward_traced(const Tensor &input, NodeId input_node, ComputationGraph &graph, Autograd &autograd)	pulsatrix::Module	inline
is_training() const	pulsatrix::Module	inline
mu_k() const	pulsatrix::RWKVModule	inline
mu_k_grad() const	pulsatrix::RWKVModule	inline
mu_r() const	pulsatrix::RWKVModule	inline
mu_r_grad() const	pulsatrix::RWKVModule	inline
mu_v() const	pulsatrix::RWKVModule	inline
mu_v_grad() const	pulsatrix::RWKVModule	inline
named_parameters() override	pulsatrix::RWKVModule	inlinevirtual
op_type() const override	pulsatrix::RWKVModule	inlinevirtual
parameters()	pulsatrix::Module	inlinevirtual
propagate_relevance(const Tensor &relevance_out, const LRPRuleConfig &config) override	pulsatrix::RWKVModule	virtual
RWKVModule(int64_t d_model, DeviceBackend *backend)	pulsatrix::RWKVModule
set_mu_k(std::initializer_list< float > values)	pulsatrix::RWKVModule
set_mu_k(const std::vector< float > &values)	pulsatrix::RWKVModule
set_mu_r(std::initializer_list< float > values)	pulsatrix::RWKVModule
set_mu_r(const std::vector< float > &values)	pulsatrix::RWKVModule
set_mu_v(std::initializer_list< float > values)	pulsatrix::RWKVModule
set_mu_v(const std::vector< float > &values)	pulsatrix::RWKVModule
set_requires_grad(bool requires_grad, const std::string &prefix="")	pulsatrix::Module	inline
set_training(bool training)	pulsatrix::Module	inlinevirtual
set_u(std::initializer_list< float > values)	pulsatrix::RWKVModule
set_u(const std::vector< float > &values)	pulsatrix::RWKVModule
set_w(std::initializer_list< float > values)	pulsatrix::RWKVModule
set_w(const std::vector< float > &values)	pulsatrix::RWKVModule
set_W_k(std::initializer_list< float > values)	pulsatrix::RWKVModule
set_W_k(const std::vector< float > &values)	pulsatrix::RWKVModule
set_W_o(std::initializer_list< float > values)	pulsatrix::RWKVModule
set_W_o(const std::vector< float > &values)	pulsatrix::RWKVModule
set_W_r(std::initializer_list< float > values)	pulsatrix::RWKVModule
set_W_r(const std::vector< float > &values)	pulsatrix::RWKVModule
set_W_v(std::initializer_list< float > values)	pulsatrix::RWKVModule
set_W_v(const std::vector< float > &values)	pulsatrix::RWKVModule
supports_lrp_rule(LRPRule rule) const	pulsatrix::Module	inlinevirtual
u() const	pulsatrix::RWKVModule	inline
u_grad() const	pulsatrix::RWKVModule	inline
w() const	pulsatrix::RWKVModule	inline
w_grad() const	pulsatrix::RWKVModule	inline
W_k() const	pulsatrix::RWKVModule	inline
W_k_grad() const	pulsatrix::RWKVModule	inline
W_o() const	pulsatrix::RWKVModule	inline
W_o_grad() const	pulsatrix::RWKVModule	inline
W_r() const	pulsatrix::RWKVModule	inline
W_r_grad() const	pulsatrix::RWKVModule	inline
W_v() const	pulsatrix::RWKVModule	inline
W_v_grad() const	pulsatrix::RWKVModule	inline
~Module()=default	pulsatrix::Module	virtual