UpdateOutputDesc

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

Updates TensorDesc of an operator output based on the output name.

Prototype

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

1
2
3
graphStatus UpdateOutputDesc(const std::string &name, const TensorDesc &tensor_desc)
graphStatus UpdateOutputDesc(const char_t *name, const TensorDesc &tensor_desc)
graphStatus UpdateOutputDesc(const uint32_t index, const TensorDesc &tensor_desc)

Parameters

Parameter

Input/Output

Description

name

Input

Name of the operator output.

tensor_desc

Input

TensorDesc object.

index

Input

Index of the operator output.

Returns

graphStatus type. GRAPH_SUCCESS on success; GRAPH_FAILED on failure.

Exception Handling

None

Restrictions

None