aimath_basic.h

Go to the documentation of this file.

 
#ifndef AIMATH_BASIC
#define AIMATH_BASIC
 
#include "core/aifes_math.h"
 
void print_aitensor(const aitensor_t *tensor);
 
void print_aiscalar(const void *scalar, const aimath_dtype_t *dtype);
 
void aimath_transpose_vector(aitensor_t *vector);
 
uint32_t aimath_tensor_elements(const aitensor_t *tensor);
 
uint32_t aimath_sizeof_dtype(const aimath_dtype_t *dtype);
 
uint32_t aimath_sizeof_tensor_data(const aitensor_t *tensor);
 
uint32_t aimath_sizeof_tensor_params(const aitensor_t *tensor);
 
uint32_t aimath_sizeof_tensor(const aitensor_t *tensor);
 
//void aimath_create_tensor_2d(uint16_t shape_0, uint16_t shape_1, aimath_dtype_t dtype, void *memory_ptr);
 
#endif // AIMATH_BASIC