pulsatrix
Loading...
Searching...
No Matches
pulsatrix::RetNetModule Member List

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

backward(const Tensor &grad_output) overridepulsatrix::RetNetModulevirtual
compute_device() const overridepulsatrix::RetNetModuleinlinevirtual
forward(const Tensor &input)pulsatrix::Moduleinline
forward_impl(const Tensor &input) overridepulsatrix::RetNetModuleprotectedvirtual
forward_traced(const Tensor &input, NodeId input_node, ComputationGraph &graph, Autograd &autograd)pulsatrix::Moduleinline
gamma() constpulsatrix::RetNetModuleinline
is_training() constpulsatrix::Moduleinline
named_parameters() overridepulsatrix::RetNetModuleinlinevirtual
op_type() const overridepulsatrix::RetNetModuleinlinevirtual
parameters()pulsatrix::Moduleinlinevirtual
propagate_relevance(const Tensor &relevance_out, const LRPRuleConfig &config) overridepulsatrix::RetNetModulevirtual
RetNetModule(int64_t d_model, int64_t key_dim, float gamma, DeviceBackend *backend)pulsatrix::RetNetModule
set_requires_grad(bool requires_grad, const std::string &prefix="")pulsatrix::Moduleinline
set_training(bool training)pulsatrix::Moduleinlinevirtual
set_W_K(std::initializer_list< float > values)pulsatrix::RetNetModule
set_W_K(const std::vector< float > &values)pulsatrix::RetNetModule
set_W_Q(std::initializer_list< float > values)pulsatrix::RetNetModule
set_W_Q(const std::vector< float > &values)pulsatrix::RetNetModule
set_W_V(std::initializer_list< float > values)pulsatrix::RetNetModule
set_W_V(const std::vector< float > &values)pulsatrix::RetNetModule
supports_lrp_rule(LRPRule rule) constpulsatrix::Moduleinlinevirtual
W_K() constpulsatrix::RetNetModuleinline
W_K_grad() constpulsatrix::RetNetModuleinline
W_Q() constpulsatrix::RetNetModuleinline
W_Q_grad() constpulsatrix::RetNetModuleinline
W_V() constpulsatrix::RetNetModuleinline
W_V_grad() constpulsatrix::RetNetModuleinline
~Module()=defaultpulsatrix::Modulevirtual