Module collenchyma_nn::frameworks::native::helper
[−]
[src]
Provides useful macros for easier NN implementation for native.
Structs
| ConvolutionConfig | |
| NormalizationConfig | |
| PoolingConfig |
Functions
| relu |
Computes the ReLU Function on the CPU |
| relu_grad |
Computes the ReLU Gradient on the CPU |
| sigmoid |
Computes the Sigmoid Function on the CPU |
| sigmoid_grad |
Computes the Sigmoid Gradient on the CPU |
| tanh |
Computes the Tanh Function on the CPU |
| tanh_grad |
Computes the Tanh Gradient on the CPU |
| write_to_memory |
Just a helper function until SharedTensor has a nice interface for writing data |