神经网络编译器配置方法、装置、计算机设备和存储介质

摘要

本申请涉及一种神经网络编译器配置方法、装置、计算机设备和存储介质。所述方法包括：获取目标神经网络模型，以及目标神经网络模型对应的初始计算图；初始计算图包括多个算子；对多个算子进行划分，得到多个算子集合；确定各算子集合分别对应的多种算子类型组合，并获取各算子类型组合对应的算子运行时间，得到各算子集合分别对应的多个算子运行时间；将多种算子类型组合中，算子运行时间最小的算子类型组合，作为各算子集合对应的目标算子类型组合；根据各算子集合对应的目标算子类型组合生成目标神经网络模型的目标计算图，根据目标计算图生成针对于目标神经网络模型的编译器配置信息。采用本方法能够提高神经网络模型编译器配置优化效果。