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

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

backward(const Tensor &grad_output) overridepulsatrix::GRUModulevirtual
bias_n() constpulsatrix::GRUModuleinline
bias_n_grad() constpulsatrix::GRUModuleinline
bias_r() constpulsatrix::GRUModuleinline
bias_r_grad() constpulsatrix::GRUModuleinline
bias_z() constpulsatrix::GRUModuleinline
bias_z_grad() constpulsatrix::GRUModuleinline
compute_device() const overridepulsatrix::GRUModuleinlinevirtual
forward(const Tensor &input)pulsatrix::Moduleinline
forward_impl(const Tensor &input) overridepulsatrix::GRUModuleprotectedvirtual
forward_traced(const Tensor &input, NodeId input_node, ComputationGraph &graph, Autograd &autograd)pulsatrix::Moduleinline
GRUModule(int64_t input_size, int64_t hidden_size, DeviceBackend *backend)pulsatrix::GRUModule
is_training() constpulsatrix::Moduleinline
named_parameters() overridepulsatrix::GRUModuleinlinevirtual
op_type() const overridepulsatrix::GRUModuleinlinevirtual
parameters()pulsatrix::Moduleinlinevirtual
propagate_relevance(const Tensor &relevance_out, const LRPRuleConfig &config) overridepulsatrix::GRUModulevirtual
set_bias_n(std::initializer_list< float > values)pulsatrix::GRUModule
set_bias_r(std::initializer_list< float > values)pulsatrix::GRUModule
set_bias_z(std::initializer_list< float > values)pulsatrix::GRUModule
set_requires_grad(bool requires_grad, const std::string &prefix="")pulsatrix::Moduleinline
set_training(bool training)pulsatrix::Moduleinlinevirtual
set_weight_hn(std::initializer_list< float > values)pulsatrix::GRUModule
set_weight_hr(std::initializer_list< float > values)pulsatrix::GRUModule
set_weight_hz(std::initializer_list< float > values)pulsatrix::GRUModule
set_weight_xn(std::initializer_list< float > values)pulsatrix::GRUModule
set_weight_xr(std::initializer_list< float > values)pulsatrix::GRUModule
set_weight_xz(std::initializer_list< float > values)pulsatrix::GRUModule
supports_lrp_rule(LRPRule rule) constpulsatrix::Moduleinlinevirtual
weight_hn() constpulsatrix::GRUModuleinline
weight_hn_grad() constpulsatrix::GRUModuleinline
weight_hr() constpulsatrix::GRUModuleinline
weight_hr_grad() constpulsatrix::GRUModuleinline
weight_hz() constpulsatrix::GRUModuleinline
weight_hz_grad() constpulsatrix::GRUModuleinline
weight_xn() constpulsatrix::GRUModuleinline
weight_xn_grad() constpulsatrix::GRUModuleinline
weight_xr() constpulsatrix::GRUModuleinline
weight_xr_grad() constpulsatrix::GRUModuleinline
weight_xz() constpulsatrix::GRUModuleinline
weight_xz_grad() constpulsatrix::GRUModuleinline
~Module()=defaultpulsatrix::Modulevirtual