GetWorkspaceSizeFn

Obtains the size of workspace.

OpRegistrationData& GetWorkspaceSizeFn(GetWorkspaceSizeFunc getWorkspaceSizeFn);

Parameter	Input/Output	Description
getWorkspaceSizeFn	Input	Callback function GetWorkspaceSizeFunc. For details, see Callback Function GetWorkspaceSizeFunc.

You can customize and implement GetWorkspaceSizeFunc class function to calculate the workspace size of an operator and export the calculation result.

Syntax

Status GetWorkspaceSizeFn(domi::Status (const ge::Operator& op, std::vector<int64_t>& size));

Parameter Description

Parameter	Input/Output	Description
op	Input	Operator object whose workspace needs to be calculated For details about the operator class, see Operator Class APIs in GE API Reference.
size	Output	Size of the workspace. The value needs to be calculated and provided by the user.

Parameter

Input/Output

Description

Input

Operator object whose workspace needs to be calculated

For details about the operator class, see Operator Class APIs in GE API Reference.

size

Output

Size of the workspace. The value needs to be calculated and provided by the user.

Parent topic: Class OpRegistrationData

Thank you very much for your feedback. We will continue working to improve the documentation.See the reply and handling status in My Cloud VOC.

The system is busy. Please try again later.

Which of the following issues have you encountered?

Content is inconsistent with the product UI

Unclear descriptions

Lack of examples or code

Incorrect steps

Can't find what I need

Lack of best practices

Feedback (optional)

0/500

Select at least one type of issue, and enter your comments or suggestions.

Enter a maximum of 500 characters.

Submit Cancel

For any further questions, feel free to contact us through the chatbot.