|
pulsatrix
|
This is the complete list of members for pulsatrix::EmbeddingModule, including all inherited members.
| backward(const Tensor &grad_output) override | pulsatrix::EmbeddingModule | virtual |
| compute_device() const | pulsatrix::Module | inlinevirtual |
| EmbeddingModule(int64_t num_embeddings, int64_t embedding_dim, DeviceBackend *backend) | pulsatrix::EmbeddingModule | |
| forward(const Tensor &input) | pulsatrix::Module | inline |
| forward_impl(const Tensor &input) override | pulsatrix::EmbeddingModule | protectedvirtual |
| forward_traced(const Tensor &input, NodeId input_node, ComputationGraph &graph, Autograd &autograd) | pulsatrix::Module | inline |
| is_training() const | pulsatrix::Module | inline |
| named_parameters() override | pulsatrix::EmbeddingModule | inlinevirtual |
| op_type() const override | pulsatrix::EmbeddingModule | inlinevirtual |
| parameters() | pulsatrix::Module | inlinevirtual |
| propagate_relevance(const Tensor &relevance_out, const LRPRuleConfig &config) override | pulsatrix::EmbeddingModule | virtual |
| set_requires_grad(bool requires_grad, const std::string &prefix="") | pulsatrix::Module | inline |
| set_training(bool training) | pulsatrix::Module | inlinevirtual |
| set_weight(std::initializer_list< float > values) | pulsatrix::EmbeddingModule | |
| set_weight(const std::vector< float > &values) | pulsatrix::EmbeddingModule | |
| supports_lrp_rule(LRPRule rule) const | pulsatrix::Module | inlinevirtual |
| weight() const | pulsatrix::EmbeddingModule | inline |
| weight_grad() const | pulsatrix::EmbeddingModule | inline |
| ~Module()=default | pulsatrix::Module | virtual |