pulsatrix::RMSNormModule Member List

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

backward(const Tensor &grad_output) override	pulsatrix::RMSNormModule	virtual
compute_device() const override	pulsatrix::RMSNormModule	inlinevirtual
forward(const Tensor &input)	pulsatrix::Module	inline
forward_impl(const Tensor &input) override	pulsatrix::RMSNormModule	protectedvirtual
forward_traced(const Tensor &input, NodeId input_node, ComputationGraph &graph, Autograd &autograd)	pulsatrix::Module	inline
gamma() const	pulsatrix::RMSNormModule	inline
gamma_grad() const	pulsatrix::RMSNormModule	inline
is_training() const	pulsatrix::Module	inline
named_parameters() override	pulsatrix::RMSNormModule	inlinevirtual
op_type() const override	pulsatrix::RMSNormModule	inlinevirtual
parameters()	pulsatrix::Module	inlinevirtual
propagate_relevance(const Tensor &relevance_out, const LRPRuleConfig &config) override	pulsatrix::RMSNormModule	virtual
RMSNormModule(int64_t num_features, DeviceBackend *backend, DeviceType device, float eps=1e-6f)	pulsatrix::RMSNormModule
RMSNormModule(int64_t num_features, DeviceBackend *backend)	pulsatrix::RMSNormModule
set_gamma(std::initializer_list< float > values)	pulsatrix::RMSNormModule
set_gamma(const std::vector< float > &values)	pulsatrix::RMSNormModule
set_requires_grad(bool requires_grad, const std::string &prefix="")	pulsatrix::Module	inline
set_training(bool training)	pulsatrix::Module	inlinevirtual
supports_lrp_rule(LRPRule rule) const	pulsatrix::Module	inlinevirtual
~Module()=default	pulsatrix::Module	virtual