dpuGetOutputTensorScale() - 1.4 English

Vitis AI User Guide (UG1414)

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

Synopsis

float dpuGetOutputTensorScale
(
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 scale value of DPU Task’s output Tensor. For each DPU output Tensor, it has one unified scale value indicating its quantization information for reformatting between data types of INT8 and FP32.

Returns

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

Include File

n2cube.h

Availability

Vitis AI v1.0