dpuGetOutputTensorChannel() - 1.1 English

Vitis AI User Guide (UG1414)

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

Synopsis

int dpuGetOutputTensorChannel
(
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 channel dimension of DPU Task’s output Tensor.

Returns

The channel dimension of Task’s output Tensor on success, or report error in case of any failure.

Include File

n2cube.h

Availability

Vitis AI v1.0