nn_i32.h File Reference

#include "float16.h"

Include dependency graph for nn_i32.h:

Go to the source code of this file.

Data Structures
struct	Tensor0D_I32
struct	Tensor1D_I32
struct	Tensor2D_I32

Functions
Tensor0D_I32 *	nn_tensor0d_i32 (int32_t data)
Tensor1D_I32 *	nn_tensor1d_i32 (size_t shape[1], const int32_t *data)
Tensor2D_I32 *	nn_tensor2d_i32 (size_t shape[2], const int32_t *data)
Tensor0D_I32 *	nn_zeros0d_i32 ()
Tensor1D_I32 *	nn_zeros1d_i32 (size_t shape[1])
Tensor2D_I32 *	nn_zeros2d_i32 (size_t shape[2])
Tensor0D_I32 *	nn_ones0d_i32 ()
Tensor1D_I32 *	nn_ones1d_i32 (size_t shape[1])
Tensor2D_I32 *	nn_ones2d_i32 (size_t shape[2])
Tensor0D_I32 *	nn_full0d_i32 (int32_t data)
Tensor1D_I32 *	nn_full1d_i32 (size_t shape[1], int32_t data)
Tensor2D_I32 *	nn_full2d_i32 (size_t shape[2], int32_t data)
Tensor0D_I32 *	nn_rand0d_i32 ()
Tensor1D_I32 *	nn_rand1d_i32 (size_t shape[1])
Tensor2D_I32 *	nn_rand2d_i32 (size_t shape[2])
void	nn_print_i32 (int32_t v, int16_t num_digits)
void	nn_print_tensor1d_i32 (const Tensor1D_I32 *tensor)
void	nn_print_tensor2d_i32 (const Tensor2D_I32 *tensor)
uint8_t	nn_equals0d_i32 (const Tensor0D_I32 *a, const Tensor0D_I32 *b)
uint8_t	nn_equals1d_i32 (const Tensor1D_I32 *a, const Tensor1D_I32 *b)
uint8_t	nn_equals2d_i32 (const Tensor2D_I32 *a, const Tensor2D_I32 *b)
void	nn_add1d_i32 (Tensor1D_I32 *y, const Tensor1D_I32 *x1, const Tensor1D_I32 *x2)
void	nn_add2d_i32 (Tensor2D_I32 *y, const Tensor2D_I32 *x1, const Tensor2D_I32 *x2)
void	nn_addscalar1d_i32 (Tensor1D_I32 *y, const Tensor1D_I32 *x, int32_t scalar)
void	nn_addscalar2d_i32 (Tensor2D_I32 *y, const Tensor2D_I32 *x, int32_t scalar)
void	nn_dot_i32 (Tensor1D_I32 *y, const Tensor1D_I32 *x1, const Tensor1D_I32 *x2)
void	nn_mm_i32 (Tensor2D_I32 *y, const Tensor2D_I32 *x1, const Tensor2D_I32 *x2)
void	nn_addmm_i32 (Tensor2D_I32 *y, const Tensor2D_I32 *c, const Tensor2D_I32 *x1, const Tensor2D_I32 *x2)
void	nn_linear_i32 (Tensor2D_I32 *y, const Tensor2D_I32 *x, const Tensor2D_I32 *weight, const Tensor1D_I32 *bias)
void	nn_relu2d_i32 (Tensor2D_I32 *y, const Tensor2D_I32 *x)