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

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

backward(const Tensor &grad_output) overridepulsatrix::LinearModulevirtual
bias() constpulsatrix::LinearModuleinline
bias_grad() constpulsatrix::LinearModuleinline
compute_device() const overridepulsatrix::LinearModuleinlinevirtual
forward(const Tensor &input)pulsatrix::Moduleinline
forward_impl(const Tensor &input) overridepulsatrix::LinearModuleprotectedvirtual
forward_traced(const Tensor &input, NodeId input_node, ComputationGraph &graph, Autograd &autograd)pulsatrix::Moduleinline
is_training() constpulsatrix::Moduleinline
LinearModule(int64_t in_features, int64_t out_features, DeviceBackend *backend, DeviceType device)pulsatrix::LinearModule
LinearModule(int64_t in_features, int64_t out_features, DeviceBackend *backend)pulsatrix::LinearModule
named_parameters() overridepulsatrix::LinearModuleinlinevirtual
op_type() const overridepulsatrix::LinearModuleinlinevirtual
parameters()pulsatrix::Moduleinlinevirtual
propagate_relevance(const Tensor &relevance_out, const LRPRuleConfig &config) overridepulsatrix::LinearModulevirtual
set_bias(std::initializer_list< float > values)pulsatrix::LinearModule
set_bias(const std::vector< float > &values)pulsatrix::LinearModule
set_requires_grad(bool requires_grad, const std::string &prefix="")pulsatrix::Moduleinline
set_training(bool training)pulsatrix::Moduleinlinevirtual
set_weight(std::initializer_list< float > values)pulsatrix::LinearModule
set_weight(const std::vector< float > &values)pulsatrix::LinearModule
supports_lrp_rule(LRPRule) const overridepulsatrix::LinearModuleinlinevirtual
weight() constpulsatrix::LinearModuleinline
weight_grad() constpulsatrix::LinearModuleinline
~Module()=defaultpulsatrix::Modulevirtual