Help Center/ Atlas 300 Application (Model 3000)/ Framework API Reference/ Non-Class Member Functions/ aicpu_run_func
Updated on 2022-03-13 GMT+08:00
View PDF
Share

aicpu_run_func

Function

Pointer to the function of an AI CPU custom operator. The function name can be customized. This function is defined in cce\customize.h.

Syntax

void (*aicpu_run_func)(opTensor_t **, void **, int32_t,

opTensor_t **, void **, int32_t, void *, rtStream_t);

Parameter Description

Parameter

Input/Output

Description

opTensor_t **

Input

Input descriptor 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 **

Input

Address of the input descriptor tensor

int32_t

Input

Number of input descriptors

opTensor_t **

Input

Output descriptor tensor

void **

Input

Address of the output descriptor tensor

int32_t

Input

Number of output descriptors

void *

Input

Attribute handle address

rtStream_t

Input

Stream identifier