Synopsis
int dpuGetOutputTensorSize
(
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 output 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 output tensor for the Node, with default value as 0.
Description
Get the size (in Byte) of DPU Task’s output Tensor.
Returns
The size of Task’s output Tensor on success, or report error in case of any failure.
See Also
Include File
n2cube.h
Availability
Vitis AI v1.0