OPERATOR REGISTRATION METHOD AND APPARATUS FOR DEEP LEARNING FRAMEWORK, DEVICE AND STORAGE MEDIUM
Patent №
US 11,625,248
Granted
2023-04-11
Filed 2022
Owner
BEIJING BAIDU NETCOM SCIENCE TECHNOLOGY CO., LTD.
Lab
—
AI components
3
ml · planning · hardware
Assignment
Recorded
Dataset
AIPD
2023_r1 edition
Application
17572140
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.
AI classification
Ownership
BEIJING BAIDU NETCOM SCIENCE TECHNOLOGY CO., LTD.
assignment · 586100130
Assignors
CHEN, WEIHANG, YANG, JIABIN, LIU, HONGYU, LAN, XIANG
On an employer assignment, the assignors are typically the inventors.