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

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

backward(const Tensor &grad_output) overridepulsatrix::LSTMModulevirtual
bias_f() constpulsatrix::LSTMModuleinline
bias_f_grad() constpulsatrix::LSTMModuleinline
bias_g() constpulsatrix::LSTMModuleinline
bias_g_grad() constpulsatrix::LSTMModuleinline
bias_i() constpulsatrix::LSTMModuleinline
bias_i_grad() constpulsatrix::LSTMModuleinline
bias_o() constpulsatrix::LSTMModuleinline
bias_o_grad() constpulsatrix::LSTMModuleinline
compute_device() const overridepulsatrix::LSTMModuleinlinevirtual
forward(const Tensor &input)pulsatrix::Moduleinline
forward_impl(const Tensor &input) overridepulsatrix::LSTMModuleprotectedvirtual
forward_traced(const Tensor &input, NodeId input_node, ComputationGraph &graph, Autograd &autograd)pulsatrix::Moduleinline
is_training() constpulsatrix::Moduleinline
LSTMModule(int64_t input_size, int64_t hidden_size, DeviceBackend *backend)pulsatrix::LSTMModule
named_parameters() overridepulsatrix::LSTMModuleinlinevirtual
op_type() const overridepulsatrix::LSTMModuleinlinevirtual
parameters()pulsatrix::Moduleinlinevirtual
propagate_relevance(const Tensor &relevance_out, const LRPRuleConfig &config) overridepulsatrix::LSTMModulevirtual
set_bias_f(std::initializer_list< float > values)pulsatrix::LSTMModule
set_bias_g(std::initializer_list< float > values)pulsatrix::LSTMModule
set_bias_i(std::initializer_list< float > values)pulsatrix::LSTMModule
set_bias_o(std::initializer_list< float > values)pulsatrix::LSTMModule
set_requires_grad(bool requires_grad, const std::string &prefix="")pulsatrix::Moduleinline
set_training(bool training)pulsatrix::Moduleinlinevirtual
set_weight_hf(std::initializer_list< float > values)pulsatrix::LSTMModule
set_weight_hg(std::initializer_list< float > values)pulsatrix::LSTMModule
set_weight_hi(std::initializer_list< float > values)pulsatrix::LSTMModule
set_weight_ho(std::initializer_list< float > values)pulsatrix::LSTMModule
set_weight_xf(std::initializer_list< float > values)pulsatrix::LSTMModule
set_weight_xg(std::initializer_list< float > values)pulsatrix::LSTMModule
set_weight_xi(std::initializer_list< float > values)pulsatrix::LSTMModule
set_weight_xo(std::initializer_list< float > values)pulsatrix::LSTMModule
supports_lrp_rule(LRPRule rule) constpulsatrix::Moduleinlinevirtual
weight_hf() constpulsatrix::LSTMModuleinline
weight_hf_grad() constpulsatrix::LSTMModuleinline
weight_hg() constpulsatrix::LSTMModuleinline
weight_hg_grad() constpulsatrix::LSTMModuleinline
weight_hi() constpulsatrix::LSTMModuleinline
weight_hi_grad() constpulsatrix::LSTMModuleinline
weight_ho() constpulsatrix::LSTMModuleinline
weight_ho_grad() constpulsatrix::LSTMModuleinline
weight_xf() constpulsatrix::LSTMModuleinline
weight_xf_grad() constpulsatrix::LSTMModuleinline
weight_xg() constpulsatrix::LSTMModuleinline
weight_xg_grad() constpulsatrix::LSTMModuleinline
weight_xi() constpulsatrix::LSTMModuleinline
weight_xi_grad() constpulsatrix::LSTMModuleinline
weight_xo() constpulsatrix::LSTMModuleinline
weight_xo_grad() constpulsatrix::LSTMModuleinline
~Module()=defaultpulsatrix::Modulevirtual