Module coaster_nn::frameworks::native::helper

Provides useful macros for easier NN implementation for native.

Structs

• ConvolutionConfig
• DropoutConfig
• NormalizationConfig
• PoolingConfig
• RnnConfig

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