TryGetInputDesc

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.

Prototype

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

graphStatus TryGetInputDesc(const std::string &name, TensorDesc &tensor_desc) const
graphStatus TryGetInputDesc(const char_t *name, TensorDesc &tensor_desc) const

Parameters

Parameter	Input/Output	Description
name	Input	Name of the operator input.
tensor_desc	Output	Current format of the operator port, which is a TensorDesc object.

Returns

graphStatus type. True indicates that the port is available and TensorDesc is successfully obtained. False indicates that the port is unavailable and the output parameter is empty. As a result, TensorDesc fails to be obtained.

Exception Handling

Exception	Description
No input corresponding to the name.	False is returned.

Restrictions

None

Parent topic: Operator