GetInputDesc

Applicability

Product	Supported or Not
Atlas A3 training products / Atlas A3 inference products	√
Atlas A2 training products / Atlas A2 inference products	√
Atlas 200I/500 A2 inference products	√
Atlas inference products	√
Atlas training products	√

Header File/Library File

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

Function Usage

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

Prototype

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

      
           TensorDesc GetInputDesc(const std::string &name) const
TensorDesc GetInputDescByName(const char_t *name) const
TensorDesc GetInputDesc(uint32_t index) const

Parameters

Parameter	Input/Output	Description
name	Input	Name of the operator input. If no operator input name is available, 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 input. If no operator input index is available, 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 input.

If no operator input name is available, 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 input.

If no operator input index is available, 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 input.

Exception Handling

None

Constraints

None

Parent topic: Operator