CANN/asc-devkit AddOutputTd函数

AddOutputTd【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit功能说明为算子增加输入Tensor的描述。函数原型ContextBuilder AddOutputTd(int32_t index, ge::DataType dtype, ge::Format originFormat, ge::Format storageFormat, gert::StorageShape storageShape)参数说明参数输入/输出说明index输入算子输出索引从0开始计数。dtype输入算子输出tensor的数据类型originFormat输入算子输出tensor原始格式storageFormat输入算子输出tensor运行时格式storageShape输入算子输出tensor的shape返回值说明当前ContextBuilder的对象约束说明调用AddOutputTd前需要调用NodeIoNum与IrInstanceNum接口调用示例gert::StorageShape x_shape {{1024, 5120}, {1024, 5120}}; gert::StorageShape output_shape {{1024, 5210}, {1024, 5210}}; context_ascendc::ContextBuilder builder; (void)builder .NodeIoNum(1, 1) .IrInstanceNum({1}) .SetOpNameType(tmpName, tmpType) .AddInputTd(0, ge::DT_FLOAT16, ge::FORMAT_ND, ge::FORMAT_ND, x_shape) .AddOutputTd(0, ge::DT_FLOAT16, ge::FORMAT_ND, ge::FORMAT_ND, output_shape)【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考