CANN昇腾算子开发套件
SetSingleOutputShape【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit功能说明设置单核上结果矩阵Output的形状。函数原型void SetSingleOutputShape(int64_t singleCo, int64_t singleDo, int64_t singleM)参数说明参数名输入/输出描述singleCo输入单核上输出通道的大小。singleDo输入单核上Output D维度大小。singleM输入单核上Output M维度大小。返回值说明无约束说明无调用示例// 实例化Conv3D Api auto ascendcPlatform platform_ascendc::PlatformAscendC(context-GetPlatformInfo()); Conv3dTilingApi::Conv3dTiling conv3dApiTiling(ascendcPlatform); conv3dApiTiling.SetSingleOutputShape(singleCo, singleDo, singleM);【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2602841.html
如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!