Baremetal-NN
Baremetal-NN API documentation
Loading...
Searching...
No Matches
nn_f16.h File Reference
#include "float16.h"
Include dependency graph for nn_f16.h:

Go to the source code of this file.

Data Structures

struct  Tensor0D_F16
 
struct  Tensor1D_F16
 
struct  Tensor2D_F16
 

Functions

static uint8_t nn_equal_f16 (float16_t golden, float16_t actual, float rel_err)
 
Tensor0D_F16nn_tensor0d_f16 (float16_t data)
 
Tensor1D_F16nn_tensor1d_f16 (size_t shape[1], const float16_t *data)
 
Tensor2D_F16nn_tensor2d_f16 (size_t shape[2], const float16_t *data)
 
Tensor0D_F16nn_zeros0d_f16 ()
 
Tensor1D_F16nn_zeros1d_f16 (size_t shape[1])
 
Tensor2D_F16nn_zeros2d_f16 (size_t shape[2])
 
Tensor0D_F16nn_ones0d_f16 ()
 
Tensor1D_F16nn_ones1d_f16 (size_t shape[1])
 
Tensor2D_F16nn_ones2d_f16 (size_t shape[2])
 
Tensor0D_F16nn_full0d_f16 (float16_t data)
 
Tensor1D_F16nn_full1d_f16 (size_t shape[1], float16_t data)
 
Tensor2D_F16nn_full2d_f16 (size_t shape[2], float16_t data)
 
Tensor0D_F16nn_rand0d_f16 ()
 
Tensor1D_F16nn_rand1d_f16 (size_t shape[1])
 
Tensor2D_F16nn_rand2d_f16 (size_t shape[2])
 
void nn_print_f16 (float16_t v, int16_t num_digits)
 
void nn_print_tensor1d_f16 (const Tensor1D_F16 *tensor)
 
void nn_print_tensor2d_f16 (const Tensor2D_F16 *tensor)
 
uint8_t nn_equals0d_f16 (const Tensor0D_F16 *a, const Tensor0D_F16 *b, float rel_err)
 
uint8_t nn_equals1d_f16 (const Tensor1D_F16 *a, const Tensor1D_F16 *b, float rel_err)
 
uint8_t nn_equals2d_f16 (const Tensor2D_F16 *a, const Tensor2D_F16 *b, float rel_err)
 
void nn_max1d_f16 (Tensor0D_F16 *y, const Tensor1D_F16 *x)
 
void nn_max2d_f16 (Tensor0D_F16 *y, const Tensor2D_F16 *x)
 
void nn_min1d_f16 (Tensor0D_F16 *y, const Tensor1D_F16 *x)
 
void nn_min2d_f16 (Tensor0D_F16 *y, const Tensor2D_F16 *x)
 
void nn_add1d_f16 (Tensor1D_F16 *y, const Tensor1D_F16 *x1, const Tensor1D_F16 *x2)
 
void nn_add2d_f16 (Tensor2D_F16 *y, const Tensor2D_F16 *x1, const Tensor2D_F16 *x2)
 
void nn_addscalar1d_f16 (Tensor1D_F16 *y, const Tensor1D_F16 *x, float16_t scalar)
 
void nn_addscalar2d_f16 (Tensor2D_F16 *y, const Tensor2D_F16 *x, float16_t scalar)
 
void nn_mul1d_f16 (Tensor1D_F16 *y, const Tensor1D_F16 *x1, const Tensor1D_F16 *x2)
 
void nn_mul2d_f16 (Tensor2D_F16 *y, const Tensor2D_F16 *x1, const Tensor2D_F16 *x2)
 
void nn_mulscalar1d_f16 (Tensor1D_F16 *y, const Tensor1D_F16 *x, float16_t scalar)
 
void nn_mulscalar2d_f16 (Tensor2D_F16 *y, const Tensor2D_F16 *x, float16_t scalar)
 
void nn_dot_f16 (Tensor1D_F16 *y, const Tensor1D_F16 *x1, const Tensor1D_F16 *x2)
 
void nn_mm_f16 (Tensor2D_F16 *y, const Tensor2D_F16 *x1, const Tensor2D_F16 *x2)
 
void nn_addmm_f16 (Tensor2D_F16 *y, const Tensor2D_F16 *c, const Tensor2D_F16 *x1, const Tensor2D_F16 *x2)
 
void nn_linear_f16 (Tensor2D_F16 *y, const Tensor2D_F16 *x, const Tensor2D_F16 *weight, const Tensor1D_F16 *bias)
 
void nn_elu2d_f16 (Tensor2D_F16 *y, const Tensor2D_F16 *x, float alpha)
 
void nn_relu2d_f16 (Tensor2D_F16 *y, const Tensor2D_F16 *x)
 
void nn_softmax1d_f16 (Tensor1D_F16 *y, const Tensor1D_F16 *x)
 
void nn_tanh2d_f16 (Tensor2D_F16 *y, const Tensor2D_F16 *x)