异构资源中神经网络计算任务的分配方法、装置和设备

摘要

本申请涉及一种异构资源中神经网络计算任务的分配方法、装置、计算机设备和存储介质。方法包括：获取神经网络的计算任务的任务信息以及异构资源的资源信息；根据任务信息以及所述资源信息确定将各子任务分配至所述异构资源执行的分配方式对应的任务处理成本；根据各任务处理成本构建有向无环图；根据有向无环图中的各所述分配路径中各所述子任务对应的任务处理成本，得到各分配路径对应的损失函数的值；根据各损失函数的值筛选出目标分配路径，本申请以子任务为分配粒度，对神经网络的计算任务进行分配，分配粒度精细，且将各子任务分配于不同种的资源，即适用于异构资源间的任务分配，适用范围相对传统技术而言更宽。