Synopsis
int dpuSetInputTensorInCHWInt8
(
DPUTask *task,
const char *nodeName,
int8_t *data,
int size,
int idx = 0
)
Arguments
- task
- The pointer to DPU Task.
- nodeName
- The pointer to DPU Node name.
- data
- The pointer to the start address of input data.
- size
- The size (in Byte) of input data to be set.
- idx
- The index of a single input tensor for the Node, with default value of 0.
Description
Set DPU Task input Tensor with data from a CPU memory block. Data is in type of INT8 and stored in Caffe Blob’s order: channel, height and weight.
Returns
0 on success, or report error in case of failure.
Include File
n2cube.h
Availability
Vitis AI v1.0