Operator registration method and apparatus for deep learning framework, device and storage medium

    公开(公告)号:US11625248B2

    公开(公告)日:2023-04-11

    申请号:US17572140

    申请日:2022-01-10

    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.

    METHOD FOR PROCESSING DATA, AND ELECTRONIC DEVICE, STORAGE MEDIUM AND PROGRAM PRODUCT

    公开(公告)号:US20230115163A1

    公开(公告)日:2023-04-13

    申请号:US17989644

    申请日:2022-11-17

    Abstract: The disclosure provides a method for processing data, and an electronic device. The method includes: obtaining first attribute information of input data and second attribute information of a computing device corresponding to the input data; selecting a target operator implementation mode from a plurality of candidate operator implementation modes based on the first attribute information and the second attribute information; determining a plurality of sub-operators included in an operator required for the input data from an operator library based on the target operator implementation mode, to generate the operator; and obtaining an operation result by performing an operation on the input data by the computing device based on the operator.

Patent Agency Ranking