Title | 基于TVM面向RISC-V设备的神经网络编译方法 |
Author | |
First Inventor | 申奥
|
Original applicant | 南方科技大学
|
First applicant | 南方科技大学
|
Address of First applicant | 518055 广东省深圳市南山区桃源街道学苑大道1088号
|
Current applicant | 南方科技大学
|
Address of Current applicant | 518055 广东省深圳市南山区桃源街道学苑大道1088号 (广东,深圳,南山区)
|
First Current Applicant | 南方科技大学
|
Address of First Current Applicant | 518055 广东省深圳市南山区桃源街道学苑大道1088号 (广东,深圳,南山区)
|
Application Number | CN202110839231.1
|
Application Date | 2021-07-23
|
Open (Notice) Number | CN113626035B
|
Date Available | 2022-11-11
|
Publication Years | 2022-11-11
|
Status of Patent | 授权
|
Legal Date | 2022-11-11
|
Subtype | 授权发明
|
SUSTech Authorship | First
|
Abstract | 本发明公开了一种基于TVM面向RISC‑V设备的神经网络编译方法,上述方法包括:基于待部署神经网络模型,通过TVM编译器获取函数文件、模型计算图文件以及模型参数文件;基于上述函数文件获取RISC‑V版本的动态链接库文件,并基于上述动态链接库文件、上述模型计算图文件以及上述模型参数文件获取RISC‑V版本的目标神经网络模型;基于上述RISC‑V版本的目标神经网络模型获取RISC‑V版本的运行时库文件、RISC‑V版本的模型计算代码以及RISC‑V版本的可执行文件。本发明方案可直接通过模型转换的方法将神经网络模型转换为能直接部署在基于RISC‑V架构的设备中的模型,可提高神经网络模型的适应性。 |
Other Abstract | 本发明公开了一种基于TVM面向RISC‑V设备的神经网络编译方法,上述方法包括:基于待部署神经网络模型,通过TVM编译器获取函数文件、模型计算图文件以及模型参数文件;基于上述函数文件获取RISC‑V版本的动态链接库文件,并基于上述动态链接库文件、上述模型计算图文件以及上述模型参数文件获取RISC‑V版本的目标神经网络模型;基于上述RISC‑V版本的目标神经网络模型获取RISC‑V版本的运行时库文件、RISC‑V版本的模型计算代码以及RISC‑V版本的可执行文件。本发明方案可直接通过模型转换的方法将神经网络模型转换为能直接部署在基于RISC‑V架构的设备中的模型,可提高神经网络模型的适应性。 |
CPC Classification Number | G06F8/41
; G06N3/02
; Y02D10/00
|
IPC Classification Number | G06F8/41
; G06N3/02
|
INPADOC Legal Status | (+PATENT GRANT)[2022-11-11][CN]
|
INPADOC Patent Family Count | 1
|
Extended Patent Family Count | 1
|
Priority date | 2021-07-23
|
Patent Agent | 朱阳波
|
Agency | 深圳市君胜知识产权代理事务所(普通合伙)
|
URL | [Source Record] |
Data Source | PatSnap
|
Document Type | Patent |
Identifier | http://kc.sustech.edu.cn/handle/2SGJ60CL/429932 |
Department | SUSTech Institute of Microelectronics |
Recommended Citation GB/T 7714 |
申奥,丁辰辰,黄耿斌,等. 基于TVM面向RISC-V设备的神经网络编译方法[P]. 2022-11-11.
|
Files in This Item: | There are no files associated with this item. |
|
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment