Invention Grant
- Patent Title: Operator registration method and apparatus for deep learning framework, device and storage medium
-
Application No.: US17572140Application Date: 2022-01-10
-
Publication No.: US11625248B2Publication Date: 2023-04-11
- Inventor: Weihang Chen , Jiabin Yang , Hongyu Liu , Xiang Lan
- Applicant: BEIJING BAIDU NETCOM SCIENCE TECHNOLOGY CO., LTD.
- Applicant Address: CN Beijing
- Assignee: BEIJING BAIDU NETCOM SCIENCE TECHNOLOGY CO., LTD.
- Current Assignee: BEIJING BAIDU NETCOM SCIENCE TECHNOLOGY CO., LTD.
- Current Assignee Address: CN Beijing
- Agency: Brooks Kushman P.C.
- Priority: CN202110540476.4 20210518
- Main IPC: G06F9/44
- IPC: G06F9/44 ; G06F9/38 ; G06F9/30 ; G06F9/54 ; G06F8/41 ; G06F8/51

Abstract:
The present disclosure provides an operator registration method and apparatus for a deep learning framework, a device and a storage medium, relates to the field of computer technologies, and specifically to the field of artificial intelligence such as deep learning. The operator registration method for a deep learning framework includes: receiving registration information provided by a user for registering operators with the deep learning framework, the registration information including: a custom calculation function, the custom calculation function being written in a manner irrelevant to the deep learning framework; building operator meta-information in the deep learning framework based on the registration information; and constructing a to-be-registered operator within the deep learning framework based on the operator meta-information, and registering the to-be-registered operator in a global operator table within the deep learning framework. The present disclosure can simplify an operator registration process.
Public/Granted literature
- US20220374238A1 OPERATOR REGISTRATION METHOD AND APPARATUS FOR DEEP LEARNING FRAMEWORK, DEVICE AND STORAGE MEDIUM Public/Granted day:2022-11-24
Information query