|
pulsatrix
|
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) override | pulsatrix::AvgPool2DModule | virtual |
| compute_device() const override | pulsatrix::AvgPool2DModule | inlinevirtual |
| forward(const Tensor &input) | pulsatrix::Module | inline |
| forward_impl(const Tensor &input) override | pulsatrix::AvgPool2DModule | protectedvirtual |
| forward_traced(const Tensor &input, NodeId input_node, ComputationGraph &graph, Autograd &autograd) | pulsatrix::Module | inline |
| is_training() const | pulsatrix::Module | inline |
| named_parameters() | pulsatrix::Module | inlinevirtual |
| op_type() const override | pulsatrix::AvgPool2DModule | inlinevirtual |
| parameters() | pulsatrix::Module | inlinevirtual |
| propagate_relevance(const Tensor &relevance_out, const LRPRuleConfig &config) override | pulsatrix::AvgPool2DModule | virtual |
| 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 |