GetInputHandleShapesAndTypes

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/inference_context.h>
  • Library file: libgraph.so

Function Usage

Obtains ShapeAndType of the operator input handle in the inference context.

Prototype

1
const std::vector<std::vector<ShapeAndType>> &GetInputHandleShapesAndTypes() const

Parameters

None

Returns

ShapeAndType of the operator input handle.

Exception Handling

None

Restrictions

None