SetInputHandleShapesAndTypes

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	√

#include <graph/inference_context.h>

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

void SetInputHandleShapesAndTypes(std::vector<std::vector<ShapeAndType>> &&shapes_and_types)

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

None

None

None

Parent topic: InferenceContext