一种针对复杂结构深度学习模型推理的算子内存分配方法

本发明涉及内存分配，具体涉及一种针对复杂结构深度学习模型推理的算子内存分配方法，统计计算图中算子的种类；确定每种算子的内存分配优先级；确定各算子的生命周期；选取未进行内存分配且内存分配优先级最高种类的算子，基于深度学习加速核NNA的存储结构和各算子的生命周期进行内存分配；重复S4，直至完成所有算子的内存分配；本发明提供的技术方案能够有效克服现有技术所存在的应用于复杂结构的深度学习模型时难以有效缩短推理时间的缺陷。