简介
概述
对于Caffe,TensorFlow等开源框架网络模型,需要使用CANN提供的模型转换工具将开源模型转换为适配昇腾AI处理器的模型。
昇腾张量编译器(Ascend Tensor Compiler,简称ATC)是异构计算架构CANN体系下的模型转换工具:
- 它可以将开源框架的网络模型(如Caffe、TensorFlow等)以及Ascend IR定义的单算子描述文件转换为昇腾AI处理器支持的离线模型。
- 模型转换过程中,ATC会进行算子调度优化、权重数据重排、内存使用优化等具体操作,对原始的深度学习模型进行进一步的调优,从而满足部署场景下的高性能需求,使其能够高效执行在昇腾AI处理器上。
本手册将重点介绍模型转换过程中所使用的参数,并介绍如何配置以及使能AIPP,如何构造单算子描述文件等功能。