摘要
本发明提供一种基于张量虚拟机的二值神经网络模型的编译方法、存储介质及计算机。所述编译方法包括:利用所述张量虚拟机加载并解析二值神经网络模型得到对应的初始Relay格式运算图,所述张量虚拟机中包括一种或多种专用TVM组合算子;遍历所述初始Relay格式运算图中的每种运算,将所述初始Relay格式运算图中的与任何一种专用TVM组合算子匹配的连续的多个运算用匹配的专用TVM组合算子替换得到改进Relay格式运算图;对所述改进Relay格式运算图进行编译得到对应的编译代码。该方法通过自定义专用TVM组合算子,不仅可以实现TVM对二值神经网络模型的支持,还可以实现二值神经网络的快捷且有效的部署。
技术关键词
编译方法
二值神经网络
加速器
神经网络处理单元
神经网络模型
格式
通道
程序
计算机
指令
参数
存储器
数据
处理器
尺寸
元素
系统为您推荐了相关专利信息
数据
时序
车辆行驶轨迹预测
长短期记忆神经网络模型
时间段
智能照明设备
室内照明系统
白色光源
光源模块
儿童房照明
健康度评估方法
信号
非标准
空间金字塔
电数字数据处理技术
山地光伏发电
光伏阵列
数据实时处理方法
轻量级深度学习
特征提取模型