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); ... }