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

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

AvgPool2DModule(int64_t kernel_h, int64_t kernel_w, DeviceBackend *backend, float eps=1e-6f)pulsatrix::AvgPool2DModule
backward(const Tensor &grad_output) overridepulsatrix::AvgPool2DModulevirtual
compute_device() const overridepulsatrix::AvgPool2DModuleinlinevirtual
forward(const Tensor &input)pulsatrix::Moduleinline
forward_impl(const Tensor &input) overridepulsatrix::AvgPool2DModuleprotectedvirtual
forward_traced(const Tensor &input, NodeId input_node, ComputationGraph &graph, Autograd &autograd)pulsatrix::Moduleinline
is_training() constpulsatrix::Moduleinline
named_parameters()pulsatrix::Moduleinlinevirtual
op_type() const overridepulsatrix::AvgPool2DModuleinlinevirtual
parameters()pulsatrix::Moduleinlinevirtual
propagate_relevance(const Tensor &relevance_out, const LRPRuleConfig &config) overridepulsatrix::AvgPool2DModulevirtual
set_requires_grad(bool requires_grad, const std::string &prefix="")pulsatrix::Moduleinline
set_training(bool training)pulsatrix::Moduleinlinevirtual
supports_lrp_rule(LRPRule rule) constpulsatrix::Moduleinlinevirtual
~Module()=defaultpulsatrix::Modulevirtual