aifes_math.h

Go to the documentation of this file.

 
#ifndef AIFES_MATH
#define AIFES_MATH
 
#include <stdint.h>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
 
#include "aifes_config.h"
 
 
typedef struct aimath_dtype aimath_dtype_t;
 
typedef struct aitensor     aitensor_t;
 
struct aimath_dtype {
    const char *name; 
    uint32_t size; 
    uint32_t tensor_params_size; 
    void (*print_aitensor)(const aitensor_t *tensor);
 
    void (*print_aiscalar)(const void *scalar);
};
 
struct aitensor {
    const aimath_dtype_t *dtype; 
    uint8_t dim; 
    uint16_t *shape; 
    void *tensor_params; 
    void *data; 
};
 
 
#endif // AIFES_MATH