DEPLOYMENT METHOD AND APPARATUS FOR HETEROGENEOUS PLATFORM BASED ON TVM COMPILER, AND TERMINAL DEVICE

摘要

Provided are a deployment method and apparatus for a heterogeneous platform based on a TVM compiler, and a terminal device. The method comprises: obtaining a topological structure diagram, network parameters and function information of each function of a deep learning network generated by a TVM compiler (S101); on the basis of information of a node in the topological structure diagram and the device type of a heterogeneous platform, dividing the node in the topological structure diagram into target device blocks corresponding to the device type (S102); performing space allocation on the target device blocks to obtain space information of the target device blocks (S103); and deploying the deep learning network to the heterogeneous platform on the basis of the topological structure diagram, the network parameters, the function information, the target device blocks and the spatial information of the target device blocks (S104). According to the method, the node is divided into target device blocks of different device types, and deployment of different devices in the heterogeneous platform is completed by means of the target device blocks of different types.