GetOutputDesc

Applicability

Product	Supported
Atlas 350 Accelerator Card	√
Atlas A3 training product / Atlas A3 inference product	√
Atlas A2 training product / Atlas A2 inference product	√
Atlas 200I/500 A2 inference product	√
Atlas inference product	√
Atlas training product	√

Header File/Library File

Header file: #include <graph/operator.h>
Library file: libgraph.so

Function Usage

Obtains TensorDesc of an operator output based on the output name or index.

Prototype

APIs taking string arguments will be deprecated in future releases. Use APIs taking non-string arguments instead.

      
           TensorDesc GetOutputDesc(const std::string &name) const
TensorDesc GetOutputDescByName(const char_t *name) const
TensorDesc GetOutputDesc(uint32_t index) const

Parameters

Parameter	Input/Output	Description
name	Input	Name of the operator output. If the operator output name is not found, the default TensorDesc object is returned. Set DataType to DT_FLOAT (indicating the float type) and set Format to FORMAT_NCHW (indicating the NCHW format).
index	Input	Index of the operator output. If the operator output index is not found, the default TensorDesc object is returned. Set DataType to DT_FLOAT (indicating the float type) and set Format to FORMAT_NCHW (indicating the NCHW format).

Parameter

Input/Output

Description

name

Input

Name of the operator output.

If the operator output name is not found, the default TensorDesc object is returned. Set DataType to DT_FLOAT (indicating the float type) and set Format to FORMAT_NCHW (indicating the NCHW format).

index

Input

Index of the operator output.

If the operator output index is not found, the default TensorDesc object is returned. Set DataType to DT_FLOAT (indicating the float type) and set Format to FORMAT_NCHW (indicating the NCHW format).

Returns

TensorDesc of the operator output

Exception Handling

None

Restrictions

None

Parent topic: Operator