下载
EN
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

CalcSize

函数功能

用于计算算子对应的ComputeNodeInfo需要预分配的内存空间大小。

ComputeNodeInfo的内存空间是平铺式的,内存依次存放ComputeNodeInfo自身的数据成员、算子IR定义输入个数的Anchor信息、实际输入个数和输出个数的编译阶段的Tensor描述信息以及属性信息。该函数的计算结果不包含属性信息所占的内存空间大小。

函数原型

static ge::graphStatus CalcSize(const size_t ir_inputs_num, const size_t inputs_num, const size_t outputs_num, size_t &total_size)

static ge::graphStatus CalcSize(const size_t ir_inputs_num, const size_t ir_outputs_num, const size_t inputs_num, const size_t outputs_num, size_t &total_size);

参数说明

参数

输入/输出

说明

ir_inputs_num

输入

算子IR原型定义的输入个数

ir_outputs_num

输入

算子IR原型定义的输出个数

inputs_num

输入

算子实际的输入个数

outputs_num

输入

算子实际的输出个数

total_size

输出

需要预分配的ComputeNodeInfo和相关信息的内存总大小,不包含RuntimeAttrs的内存大小

返回值说明

返回值为graphStatus类型(uint32_t),其不同的状态取值说明如下:

状态

说明

ge::GRAPH_SUCCESS

0

执行成功

ge::GRAPH_PARAM_INVALID

50331649

执行失败,参数无效,校验未通过

约束说明

无。

调用示例

auto ir_input_num = node->GetOpDesc()->GetIrInputs().size();
auto input_num = node->GetInDataNodesAndAnchors().size();
auto output_num = node->GetAllOutDataAnchorsSize();
GE_ASSERT_SUCCESS(ComputeNodeInfo::CalcSize(ir_input_num, input_num, output_num, total_size));
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词