|
static uint8_t | nn_equal_f16 (float16_t golden, float16_t actual, float rel_err) |
|
Tensor0D_F16 * | nn_tensor0d_f16 (float16_t data) |
|
Tensor1D_F16 * | nn_tensor1d_f16 (size_t shape[1], const float16_t *data) |
|
Tensor2D_F16 * | nn_tensor2d_f16 (size_t shape[2], const float16_t *data) |
|
Tensor0D_F16 * | nn_zeros0d_f16 () |
|
Tensor1D_F16 * | nn_zeros1d_f16 (size_t shape[1]) |
|
Tensor2D_F16 * | nn_zeros2d_f16 (size_t shape[2]) |
|
Tensor0D_F16 * | nn_ones0d_f16 () |
|
Tensor1D_F16 * | nn_ones1d_f16 (size_t shape[1]) |
|
Tensor2D_F16 * | nn_ones2d_f16 (size_t shape[2]) |
|
Tensor0D_F16 * | nn_full0d_f16 (float16_t data) |
|
Tensor1D_F16 * | nn_full1d_f16 (size_t shape[1], float16_t data) |
|
Tensor2D_F16 * | nn_full2d_f16 (size_t shape[2], float16_t data) |
|
Tensor0D_F16 * | nn_rand0d_f16 () |
|
Tensor1D_F16 * | nn_rand1d_f16 (size_t shape[1]) |
|
Tensor2D_F16 * | nn_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) |
|