通过指定的输出索引,获取需要清理的输出内存的大小。
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); ... }