Implementation of a neural network builder for constructing and registering neural network layers. More...
#include "velox/ml_functions/functions.h"Go to the source code of this file.
Classes | |
| class | NNBuilder |
| Implements a neural network builder for constructing and registering neural network layers. More... | |
Enumerations | |
| enum | Activation { RELU , SOFTMAX , NONE } |
| Enumeration of activation functions supported by the neural network builder. More... | |
Implementation of a neural network builder for constructing and registering neural network layers.
| enum Activation |
Enumeration of activation functions supported by the neural network builder.
| Enumerator | |
|---|---|
| RELU | Rectified Linear Unit (ReLU) activation. |
| SOFTMAX | Softmax activation. |
| NONE | No activation. |