GetInputDesc

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 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 the operator input 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 input. If the operator input 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 input.

If the operator input 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 input.

If the operator input 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 input

Exception Handling

None

Restrictions

None

Parent topic: Operator