GetWorkspaceSizes

函数功能

获取workspace sizes指针。

函数原型

size_t *GetWorkspaceSizes(const size_t workspace_count)；

参数说明

返回值说明

workspace sizes指针

约束说明

传入的workspace个数不可以超过编译时指定的最大workspace个数。

调用示例

ge::graphStatus Tiling4XXX(TilingContext* context) {
  auto ws = context->GetWorkspaceSizes(5);
  // ...
}