|
pulsatrix
|
This is the complete list of members for pulsatrix::RetNetModule, including all inherited members.
| backward(const Tensor &grad_output) override | pulsatrix::RetNetModule | virtual |
| compute_device() const override | pulsatrix::RetNetModule | inlinevirtual |
| forward(const Tensor &input) | pulsatrix::Module | inline |
| forward_impl(const Tensor &input) override | pulsatrix::RetNetModule | protectedvirtual |
| forward_traced(const Tensor &input, NodeId input_node, ComputationGraph &graph, Autograd &autograd) | pulsatrix::Module | inline |
| gamma() const | pulsatrix::RetNetModule | inline |
| is_training() const | pulsatrix::Module | inline |
| named_parameters() override | pulsatrix::RetNetModule | inlinevirtual |
| op_type() const override | pulsatrix::RetNetModule | inlinevirtual |
| parameters() | pulsatrix::Module | inlinevirtual |
| propagate_relevance(const Tensor &relevance_out, const LRPRuleConfig &config) override | pulsatrix::RetNetModule | virtual |
| 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::Module | inline |
| set_training(bool training) | pulsatrix::Module | inlinevirtual |
| 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) const | pulsatrix::Module | inlinevirtual |
| W_K() const | pulsatrix::RetNetModule | inline |
| W_K_grad() const | pulsatrix::RetNetModule | inline |
| W_Q() const | pulsatrix::RetNetModule | inline |
| W_Q_grad() const | pulsatrix::RetNetModule | inline |
| W_V() const | pulsatrix::RetNetModule | inline |
| W_V_grad() const | pulsatrix::RetNetModule | inline |
| ~Module()=default | pulsatrix::Module | virtual |