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

GetGetWorkspaceSizeFunc

Function

Obtains the callback function for the workspace size.

Syntax

domi::GetWorkspaceSizeFunc GetGetWorkspaceSizeFunc(const std::string& op_type);

Parameter Description

Parameter

Input/Output

Description

op_type

Input

Operator type