tensor.h File Reference

#include <stddef.h>
#include <stdint.h>
#include <stdlib.h>
#include <stdio.h>
#include <assert.h>
#include <string.h>
#include "float16.h"

Include dependency graph for tensor.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Data Structures
struct	Tensor

Macros
#define	MAX_DIMS   4

Enumerations
enum	DataType {   DTYPE_U8 , DTYPE_I8 , DTYPE_U16 , DTYPE_I16 ,   DTYPE_U32 , DTYPE_I32 , DTYPE_F16 , DTYPE_F32 }

Functions
static size_t	NN_sizeof (DataType dtype)
static const char *	NN_get_datatype_name (DataType dtype)
static uint8_t	NN_is_scalar (Tensor *tensor)
static uint8_t	NN_is_vector (Tensor *tensor)
static uint8_t	NN_is_matrix (Tensor *tensor)
static uint8_t	NN_is_3d (Tensor *tensor)
static uint8_t	NN_is_4d (Tensor *tensor)
static void	NN_free_tensor_data (Tensor *tensor)
static void	NN_delete_tensor (Tensor *tensor)