aimath_q31_default.h

Go to the documentation of this file.

 
#ifndef AIMATH_Q31_DEFAULT
#define AIMATH_Q31_DEFAULT
 
#define SHIFT_EQUALIZATION
 
#include <stdint.h>
#include <math.h>
#include <stdlib.h>
 
#include "basic/base/aimath/aimath_q31.h"
 
void aimath_q31_default_linear32(const aitensor_t *a, const aitensor_t *b, const aitensor_t *c, aitensor_t *result); //TODO Rename to aimath_q31_default_linear() (without 32) ?
 
void aimath_q31_default_mat_mul(const aitensor_t *a, const aitensor_t *b, aitensor_t *result);
 
void aimath_q31_default_multiply(const aitensor_t *a, const aitensor_t *b, aitensor_t *result);
 
void aimath_q31_default_scalar_mul(const void *scalar, const aitensor_t *a, aitensor_t *result);
 
void aimath_q31_default_tensor_add_different_shift(const aitensor_t *a, const aitensor_t *b, aitensor_t *result);
 
void aimath_q31_default_tensor_add_same_shift(const aitensor_t *a, const aitensor_t *b, aitensor_t *result);
 
void aimath_q31_default_tensor_sub_different_shift(const aitensor_t *a, const aitensor_t *b, aitensor_t *result);
 
void aimath_q31_default_tensor_sub_same_shift(const aitensor_t *a, const aitensor_t *b, aitensor_t *result);
 
void aimath_q31_default_copy_tensor(const aitensor_t *from, aitensor_t *to);
 
void aimath_q31_default_transpose_vector(aitensor_t *vector);
 
void aimath_q31_default_norm_squared(const aitensor_t *x, void *result);
 
void aimath_q31_default_tensor_sqrt(const aitensor_t *x, aitensor_t *result);
 
void aimath_q31_default_sigmoid(const aitensor_t *x, aitensor_t *result);
 
void aimath_q31_default_d_sigmoid(const aitensor_t *sigmoid_x, aitensor_t *result);
 
void aimath_q31_default_relu(const aitensor_t *x, aitensor_t *result);
 
void aimath_q31_default_d_relu(const aitensor_t *x, aitensor_t *result);
 
void aimath_q31_default_leaky_relu(const aitensor_t *x, const void *alpha, aitensor_t *result);
 
void aimath_q31_default_d_leaky_relu(const aitensor_t *x, const void *alpha, aitensor_t *result);
 
void aimath_q31_default_tanh(const aitensor_t *x, aitensor_t *result);
 
void aimath_q31_default_d_tanh(const aitensor_t *tanh_x, aitensor_t *result);
 
void aimath_q31_default_softsign(const aitensor_t *x, aitensor_t *result);
 
void aimath_q31_default_d_softsign(const aitensor_t *x, aitensor_t *result);
 
void aimath_q31_default_softmax(const aitensor_t *x, aitensor_t *result);
 
void aimath_q31_default_elu(const aitensor_t *x, const void *alpha, aitensor_t *result);
 
void aimath_q31_default_d_elu(const aitensor_t *x, const void *alpha, aitensor_t *result);
 
void aimath_q31_default_zero_tensor(aitensor_t *tensor);
 
void aimath_q31_default_init_zeros(aitensor_t *tensor);
 
void aimath_q31_default_tensor_init_uniform(aitensor_t *tensor, float from, float to);
 
void aimath_q31_default_init_glorot_uniform(aitensor_t *tensor);
 
void aimath_q31_default_init_glorot_uniform_cdim(aitensor_t *tensor, int8_t cin_axis, int8_t cout_axis);
 
void aimath_q31_default_init_he_uniform(aitensor_t *tensor);
 
void aimath_q31_default_init_he_uniform_cdim(aitensor_t *tensor, int8_t cout_axis);
 
int64_t aimath_q31_default_sqrt(int64_t x);
 
void aimath_q31_default_sum_channelwise(const aitensor_t *x, int8_t channel_axis, aitensor_t *result);
 
// TODO: docs
void aimath_q31_default_mse_gradients_sum(const aitensor_t *predicted, const aitensor_t *target, aitensor_t *result);
 
void aimath_q31_default_mse_gradients_mean(const aitensor_t *predicted, const aitensor_t *target, aitensor_t *result);
 
void aimath_q31_default_mse_loss_sum(const aitensor_t *predicted, const aitensor_t *target, void *result);
 
void aimath_q31_default_mse_loss_mean(const aitensor_t *predicted, const aitensor_t *target, void *result);
#endif // AIMATH_Q31_DEFAULT