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 |