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

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

backward(const Tensor &grad_output) overridepulsatrix::Conv2DModulevirtual
bias() constpulsatrix::Conv2DModuleinline
bias_grad() constpulsatrix::Conv2DModuleinline
compute_device() const overridepulsatrix::Conv2DModuleinlinevirtual
Conv2DModule(int64_t in_channels, int64_t out_channels, int64_t kernel_h, int64_t kernel_w, DeviceBackend *backend, int64_t stride=1, int64_t padding=0)pulsatrix::Conv2DModule
forward(const Tensor &input)pulsatrix::Moduleinline
forward_impl(const Tensor &input) overridepulsatrix::Conv2DModuleprotectedvirtual
forward_traced(const Tensor &input, NodeId input_node, ComputationGraph &graph, Autograd &autograd)pulsatrix::Moduleinline
is_training() constpulsatrix::Moduleinline
kernel() constpulsatrix::Conv2DModuleinline
kernel_grad() constpulsatrix::Conv2DModuleinline
named_parameters() overridepulsatrix::Conv2DModuleinlinevirtual
op_type() const overridepulsatrix::Conv2DModuleinlinevirtual
padding() constpulsatrix::Conv2DModuleinline
parameters()pulsatrix::Moduleinlinevirtual
propagate_relevance(const Tensor &relevance_out, const LRPRuleConfig &config) overridepulsatrix::Conv2DModulevirtual
set_bias(std::initializer_list< float > values)pulsatrix::Conv2DModule
set_bias(const std::vector< float > &values)pulsatrix::Conv2DModule
set_kernel(std::initializer_list< float > values)pulsatrix::Conv2DModule
set_kernel(const std::vector< float > &values)pulsatrix::Conv2DModule
set_requires_grad(bool requires_grad, const std::string &prefix="")pulsatrix::Moduleinline
set_training(bool training)pulsatrix::Moduleinlinevirtual
stride() constpulsatrix::Conv2DModuleinline
supports_lrp_rule(LRPRule) const overridepulsatrix::Conv2DModuleinlinevirtual
~Module()=defaultpulsatrix::Modulevirtual