CalcSize【免费下载链接】metadefAscend Metadata Definition项目地址: https://gitcode.com/cann/metadef函数功能用于计算算子对应的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_SUCCESS0执行成功。ge::GRAPH_PARAM_INVALID50331649执行失败参数无效校验未通过。约束说明无。调用示例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));【免费下载链接】metadefAscend Metadata Definition项目地址: https://gitcode.com/cann/metadef创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考