 |
NeuZephyr
Simple DL Framework
|
|
NeuZephyr
Simple DL Framework
|
This is the complete list of members for nz::Model, including all inherited members.
| Add(Node *lhs, Node *rhs) | nz::Model | protected |
| AvgPool2d(Node *input, Tensor::size_type poolSize, Tensor::size_type stride, Tensor::size_type padding=0) | nz::Model | protected |
| backward() | nz::Model | |
| BCELoss(Node *input, Node *target) | nz::Model | protected |
| Bias(Node *input) | nz::Model | protected |
| Col2Img(Node *input, Tensor::size_type outputHeight, Tensor::size_type outputWidth) | nz::Model | protected |
| Conv2d(Node *input, Tensor::size_type outChannels, Tensor::size_type kernelHeight, Tensor::size_type kernelWidth, Tensor::size_type stride, Tensor::size_type padding, bool bias=true) | nz::Model | protected |
| defaultOutput(Node *input) | nz::Model | protected |
| ELU(Node *input, float alpha=1.0f) | nz::Model | protected |
| forward() | nz::Model | |
| getLoss() const | nz::Model | |
| GlobalAvgPool2d(Node *input) | nz::Model | protected |
| GlobalMaxPool2d(Node *input) | nz::Model | protected |
| HardSigmoid(Node *input, float alpha=0.2f, float beta=0.5f) | nz::Model | protected |
| HardSwish(Node *input, float alpha=0.2f, float beta=0.5f) | nz::Model | protected |
| Img2Col(Node *input, Tensor::size_type kernelHeight, Tensor::size_type kernelWidth, Tensor::size_type stride, Tensor::size_type padding) | nz::Model | protected |
| LeakyReLU(Node *input, float alpha=0.01f) | nz::Model | protected |
| Linear(Node *input, size_t outSize) | nz::Model | protected |
| MaxPool2d(Node *input, Tensor::size_type poolSize, Tensor::size_type stride, Tensor::size_type padding=0) | nz::Model | protected |
| Model() | nz::Model | |
| MSELoss(Node *input, Node *target) | nz::Model | protected |
| Mul(Node *lhs, Node *rhs) | nz::Model | protected |
| operator<<(std::ostream &os, Model &model) | nz::Model | related |
| ReLU(Node *input) | nz::Model | protected |
| Reshape(Node *input, const Tensor::shape_type &shape) | nz::Model | protected |
| Sigmoid(Node *input) | nz::Model | protected |
| Softmax(Node *input) | nz::Model | protected |
| Sub(Node *lhs, Node *rhs) | nz::Model | protected |
| Swish(Node *input) | nz::Model | protected |
| Tanh(Node *input) | nz::Model | protected |
| TargetExpand(Node *input, const Tensor::shape_type &shape) | nz::Model | protected |
| update(opt::Optimizer *optimizer) const | nz::Model | |
| ~Model() | nz::Model |
1.12.0