|
pulsatrix
|
This is the complete list of members for pulsatrix::GroupNormModule, including all inherited members.
| backward(const Tensor &grad_output) override | pulsatrix::GroupNormModule | virtual |
| beta() const | pulsatrix::GroupNormModule | inline |
| beta_grad() const | pulsatrix::GroupNormModule | inline |
| compute_device() const override | pulsatrix::GroupNormModule | inlinevirtual |
| forward(const Tensor &input) | pulsatrix::Module | inline |
| forward_impl(const Tensor &input) override | pulsatrix::GroupNormModule | protectedvirtual |
| forward_traced(const Tensor &input, NodeId input_node, ComputationGraph &graph, Autograd &autograd) | pulsatrix::Module | inline |
| gamma() const | pulsatrix::GroupNormModule | inline |
| gamma_grad() const | pulsatrix::GroupNormModule | inline |
| GroupNormModule(int64_t num_groups, int64_t num_channels, DeviceBackend *backend, DeviceType device, float eps=1e-6f) | pulsatrix::GroupNormModule | |
| GroupNormModule(int64_t num_groups, int64_t num_channels, DeviceBackend *backend) | pulsatrix::GroupNormModule | |
| is_training() const | pulsatrix::Module | inline |
| named_parameters() override | pulsatrix::GroupNormModule | inlinevirtual |
| op_type() const override | pulsatrix::GroupNormModule | inlinevirtual |
| parameters() | pulsatrix::Module | inlinevirtual |
| propagate_relevance(const Tensor &relevance_out, const LRPRuleConfig &config) override | pulsatrix::GroupNormModule | virtual |
| set_beta(std::initializer_list< float > values) | pulsatrix::GroupNormModule | |
| set_beta(const std::vector< float > &values) | pulsatrix::GroupNormModule | |
| set_gamma(std::initializer_list< float > values) | pulsatrix::GroupNormModule | |
| set_gamma(const std::vector< float > &values) | pulsatrix::GroupNormModule | |
| set_requires_grad(bool requires_grad, const std::string &prefix="") | pulsatrix::Module | inline |
| set_training(bool training) | pulsatrix::Module | inlinevirtual |
| supports_lrp_rule(LRPRule rule) const | pulsatrix::Module | inlinevirtual |
| ~Module()=default | pulsatrix::Module | virtual |