神经网络的动态任务分配

    公开(公告)号:CN110678846A

    公开(公告)日:2020-01-10

    申请号:CN201880035228.0

    申请日:2018-04-24

    申请人: 苹果公司

    IPC分类号: G06F9/50 G06N3/063

    摘要: 本主题技术为神经网络模型的动态任务分配作准备。本主题技术确定在神经网络模型的节点处执行的操作。本主题技术至少部分地基于目标平台的硬件能力来分配注释以指示操作是在CPU上还是在GPU上更好地执行。本主题技术确定神经网络模型是否包括第二层。响应于确定该神经网络模型包括第二层,对于该神经网络模型的该第二层的每个节点,本主题技术确定在节点处执行的第二操作。另外,本主题技术至少部分地基于目标平台的硬件能力来分配第二注释以指示该第二操作是在CPU上还是在GPU上更好地执行。