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

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

backward(const Tensor &grad_output) overridepulsatrix::BatchNormModulevirtual
BatchNormFoldpulsatrix::BatchNormModulefriend
BatchNormModule(int64_t num_channels, DeviceBackend *backend, DeviceType device, float eps=1e-6f, float momentum=0.1f)pulsatrix::BatchNormModule
BatchNormModule(int64_t num_channels, DeviceBackend *backend)pulsatrix::BatchNormModule
beta() constpulsatrix::BatchNormModuleinline
beta_grad() constpulsatrix::BatchNormModuleinline
compute_device() const overridepulsatrix::BatchNormModuleinlinevirtual
forward(const Tensor &input)pulsatrix::Moduleinline
forward_impl(const Tensor &input) overridepulsatrix::BatchNormModuleprotectedvirtual
forward_traced(const Tensor &input, NodeId input_node, ComputationGraph &graph, Autograd &autograd)pulsatrix::Moduleinline
gamma() constpulsatrix::BatchNormModuleinline
gamma_grad() constpulsatrix::BatchNormModuleinline
is_training() constpulsatrix::Moduleinline
named_parameters() overridepulsatrix::BatchNormModuleinlinevirtual
op_type() const overridepulsatrix::BatchNormModuleinlinevirtual
parameters()pulsatrix::Moduleinlinevirtual
propagate_relevance(const Tensor &relevance_out, const LRPRuleConfig &config) overridepulsatrix::BatchNormModulevirtual
running_mean() constpulsatrix::BatchNormModuleinline
running_var() constpulsatrix::BatchNormModuleinline
set_beta(std::initializer_list< float > values)pulsatrix::BatchNormModule
set_beta(const std::vector< float > &values)pulsatrix::BatchNormModule
set_gamma(std::initializer_list< float > values)pulsatrix::BatchNormModule
set_gamma(const std::vector< float > &values)pulsatrix::BatchNormModule
set_requires_grad(bool requires_grad, const std::string &prefix="")pulsatrix::Moduleinline
set_running_mean(const std::vector< float > &values)pulsatrix::BatchNormModule
set_running_var(const std::vector< float > &values)pulsatrix::BatchNormModule
set_training(bool training)pulsatrix::Moduleinlinevirtual
supports_lrp_rule(LRPRule rule) constpulsatrix::Moduleinlinevirtual
~Module()=defaultpulsatrix::Modulevirtual