SetOutputHandleShapesAndTypes

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

#include <graph/inference_context.h>

Function Usage

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

Prototype

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

Constraints

None