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