aicpu_run_func函数
函数功能
AI CPU自定义算子的函数指针,函数名称可以自定义。该函数在“cce\customize.h”中定义。
函数原型
void (*aicpu_run_func)(opTensor_t **, void **, int32_t,
opTensor_t **, void **, int32_t, void *, rtStream_t);
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
opTensor_t ** |
输入 |
输入描述符Tensor。 typedef struct tagOpTensor { // real dim info opTensorFormat_t format; //char align1[4]; opDataType_t data_type; //char align2[4]; int32_t dim_cnt; int32_t mm; //lint !e148 int32_t dim[CC_DEVICE_DIM_MAX]; //lint !e148 } opTensor_t; |
void ** |
输入 |
输入描述符Tensor的地址 |
int32_t |
输入 |
输入描述符个数 |
opTensor_t ** |
输入 |
输出描述符Tensor |
void ** |
输入 |
输出描述符Tensor的地址 |
int32_t |
输入 |
输出描述符个数 |
void * |
输入 |
属性handle地址 |
rtStream_t |
输入 |
stream 标识 |