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