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.
See Also
Include File
n2cube.h
Availability
Vitis AI v1.0