news 2026/6/20 17:30:57

CANN/GE:获取模型输入大小

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
CANN/GE:获取模型输入大小

aclmdlGetInputSizeByIndex

【免费下载链接】geGE(Graph Engine)是面向昇腾的图编译器和执行器,提供了计算图优化、多流并行、内存复用和模型下沉等技术手段,加速模型执行效率,减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力,并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge

产品支持情况

产品

是否支持

Atlas A3 训练系列产品 / Atlas A3 推理系列产品

Atlas A2 训练系列产品 / Atlas A2 推理系列产品

功能说明

根据模型描述信息获取指定输入的大小,单位为Byte。

函数原型

size_t aclmdlGetInputSizeByIndex(aclmdlDesc *modelDesc, size_t index)

参数说明

参数名

输入/输出

说明

modelDesc

输入

aclmdlDesc类型的指针。

需提前调用aclmdlCreateDesc接口创建aclmdlDesc类型的数据。

index

输入

指定获取第几个输入的大小,index值从0开始。

返回值说明

针对动态Batch、动态分辨率(宽高)的场景,返回最大档位对应的输入的大小;静态场景下,返回指定输入的大小。单位是Byte。

约束说明

如果模型输入的Shape是动态的、且维度的取值为-1(表示此维度可以使用>=1的任意取值),则通过本接口获取的大小为0,用户需根据实际数据占用的内存大小来申请内存。

【免费下载链接】geGE(Graph Engine)是面向昇腾的图编译器和执行器,提供了计算图优化、多流并行、内存复用和模型下沉等技术手段,加速模型执行效率,减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力,并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/6/20 17:22:19

05AB1E高级技巧:向量化操作与压缩字符串的完整指南

05AB1E高级技巧:向量化操作与压缩字符串的完整指南 【免费下载链接】05AB1E A concise stack-based golfing language 项目地址: https://gitcode.com/gh_mirrors/05/05AB1E 05AB1E是一款简洁的栈式高尔夫语言,以其高效的代码编写能力受到开发者青…

作者头像 李华
网站建设 2026/6/20 17:13:07

为什么传统监控会失效?构建下一代分布式数据库性能洞察体系

为什么传统监控会失效?构建下一代分布式数据库性能洞察体系 【免费下载链接】oceanbase The Fastest Distributed Database for Transactional, Analytical, and AI Workloads. 项目地址: https://gitcode.com/GitHub_Trending/oc/oceanbase 在分布式数据库的…

作者头像 李华
网站建设 2026/6/20 17:06:57

三分钟学会专业歌词同步:零基础制作精准LRC歌词的完整指南

三分钟学会专业歌词同步:零基础制作精准LRC歌词的完整指南 【免费下载链接】lrc-maker 歌词滚动姬|可能是你所能见到的最好用的歌词制作工具 项目地址: https://gitcode.com/gh_mirrors/lr/lrc-maker 想象一下这样的场景:你正沉浸在音…

作者头像 李华
网站建设 2026/6/20 17:04:48

嵌入式GUI开发实战:emWin文本显示与SPY调试工具深度解析

1. 项目概述:嵌入式GUI开发中的文本显示与调试实战在嵌入式系统开发领域,图形用户界面(GUI)是连接用户与设备的核心桥梁。无论是工业控制面板上的参数设置,还是智能家居中控屏的交互反馈,清晰、流畅的文本信…

作者头像 李华
网站建设 2026/6/20 17:02:57

ARM7嵌入式开发实战:OSEKturbo OS环境搭建、任务调度与事件机制详解

1. 从零开始:理解 OSEK/VDX 与 OSEKturbo OS如果你正在开发基于 ARM7 的嵌入式系统,尤其是汽车电子控制单元(ECU)或对实时性有苛刻要求的工业控制器,那么“实时操作系统”这个词一定不会陌生。它不是 Linux 那样的通用…

作者头像 李华