模型转换方法以及模型转换设备、存储介质

摘要

本申请公开了一种模型转换方法以及模型转换设备、存储介质，模型转换方法包括：基于网络模型中各计算节点输入端和输出端的数据块的形状，在目标平台中确定各计算节点的初始数据排布格式，并获取确定过程花费的第一耗时；依次判断相邻连接的两个计算节点的初始数据排布格式是否一致，若不一致，在相邻的两个计算节点之间添加转换算子，利用转换算子对相邻的两个计算节点的初始数据排布格式进行转换，并获取转换算子的转换耗时；最后，基于第一耗时和转换耗时的相关性，确定网络模型中所有计算节点的连接结构。通过上述方式，本申请提供模型转换方法可以通过确认网络模型中所有计算节点的连接结构，从而获得一种对目标平台来说性能最优的模型。