Updated on 2022-03-13 GMT+08:00

GetInferShapeFunc

Function

Obtains the inference shape callback function.

Syntax

domi::InferShapeFunc GetInferShapeFunc(const std::string& op_type);

Parameter Description

Parameter

Input/Output

Description

op_type

Input

Operator type