TryGetInputDesc

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.

Prototype

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

1
2
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.

Constraints

None