Synopsis
int dpuSetInputTensorInCHWFP32
(
DPUTask *task,
const char *nodeName,
float *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’s input Tensor with data from a CPU memory block. Data is in type of 32-bit-float and stored in DPU Tensor’s order: height, weight and channel.
Returns
0 on success, or report error in case of failure.
Include File
n2cube.h
Availability
Vitis AI v1.0