dpuGetInputTensorSize() - 1.4 English

Vitis AI User Guide (UG1414)

Document ID
UG1414
Release Date
2021-07-22
Version
1.4 English

Synopsis

int dpuGetInputTensorSize
(
DPUTask *task,
const char*nodeName,
int idx = 0
    );

Arguments

task
The pointer to DPU Task.
nodeName
The pointer to DPU Node’s name.
Note:

The available names of one DPU Kernel’s or Task’s input Node are listed out after a neural network is compiled by VAI_C. If invalid Node name specified, failure message is reported.

idx

The index of a single input tensor for the Node, with default value as 0.

Description

Get the size (in Byte) of DPU Task’s input Tensor.

Returns

The size of Task’s input Tensor on success, or report error in case of any failure.

Include File

n2cube.h

Availability

Vitis AI v1.0