GetCleanOutputSize
函数功能
通过指定的输出索引,获取需要清理的输出内存的大小。
函数原型
uint64_t GetCleanOutputSize(size_t index)
参数说明
参数 |
输入/输出 |
说明 |
|---|---|---|
index |
输入 |
算子的输出索引,从0开始计数。 |
返回值说明
需要清理的输出内存的大小。
约束说明
无
调用示例
ge::graphStatus TilingForDynamicAtomicAddrClean(TilingContext *context) {
auto compute_node_info = context->GetComputeNodeInfo();
size_t clean_tensor_num = compute_node_info->GetInputsNum() - 1;
for (; idx < clean_tensor_num; ++idx) {
auto tensor_size = atomic_clean_context->GetCleanOutputSize(idx);
...
}