更新时间:2021-03-18 GMT+08:00
分享

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 标识

分享:

    相关文档

    相关产品