SetOutputHandleShapesAndTypes

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

Sets ShapeAndType of the operator output handle in the inference context.

Prototype

void SetOutputHandleShapesAndTypes(const std::vector<std::vector<ShapeAndType>> &shapes_and_types)
void SetOutputHandleShapesAndTypes(std::vector<std::vector<ShapeAndType>> &&shapes_and_types)

Parameters

Parameter	Input/Output	Description
shapes_and_types	Input	ShapeAndType of the operator output handle.

Returns

None

Exception Handling

None

Restrictions

None

Parent topic: InferenceContext