基于有向无环图的神经网络模型计算量的评估方法及设备

AITNT
正文
推荐专利
基于有向无环图的神经网络模型计算量的评估方法及设备
申请号:CN202411856396
申请日期:2024-12-17
公开号:CN119849545A
公开日期:2025-04-18
类型:发明专利
摘要
本申请公开了一种基于有向无环图的神经网络模型计算量的评估方法及设备,应用于计算机技术领域,所述方法包括判断神经网络模型是否满足有向无环条件,满足则获取其算子拓扑序列;判断神经网络模型是否为全static shape神经网络模型,若是,则根据其算子拓扑序列,计算每一算子的操作数;对每一算子的操作数进行累加求和,得出神经网络模型的计算量。以此方式,可以在模型部署前准确地确定计算量需求,依据准确的计算量需求提前分配芯片资源,避免因资源过度配置导致的浪费或资源不足引发的效率低下问题。
技术关键词
神经网络模型 序列 数据 元素 计算机 数值 评估系统 处理器通信 指令 可读存储介质 资源 存储器 通道 电子设备 芯片 尺寸
添加客服微信openai178,进AITNT官方交流群
驱动智慧未来:提供一站式AI转型解决方案
沪ICP备2023015588号