算子适配插件实现(TensorFlow/Caffe/ONNX)

开发者需要进行算子适配插件的开发,实现将第三方框架(TensorFlow/Caffe/ONNX)网络中的算子进行解析并映射成昇腾AI处理器中的算子。

TensorFlow/Caffe

MindStudio在“framework/tf_plugin/tensorflow_add_plugin.cc”(以TensorFlow为例)文件已自动生成了Add算子的插件代码。

ONNX

MindStudio在“framework/onnx_plugin/Add_plugin.cc”文件已自动生成了Add算子的插件代码。