GetWorkspaceSizes

函数功能

获取workspace sizes指针，workspace大小以字节为单位。

函数原型

size_t *GetWorkspaceSizes(const size_t workspace_count)

参数说明

返回值说明

workspace sizes指针。

约束说明

无。

调用示例

ge::graphStatus Tiling4XXX(TilingContext* context) {
    auto ws = context->GetWorkspaceSizes(5);
    if （ws == nullptr） {
        return ge::GRAPH_FAILED;
    }
    // ...
}