dpuSetInputTensorInCHWFP32() - 1.1 English

Vitis AI User Guide (UG1414)

Document ID
UG1414
Release Date
2020-03-23
Version
1.1 English

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