问题描述:
大家好,我是mindspore lite的新手,想研究一下lite的源码。在阅读到lite读取和解析模型的部分遇到问题。
从lite/examples,可以知道模型是通过Mindspore::Model::Build()来读取和解析模型的。最后看到LiteModel::ConstructModel就疑惑了,没有找到解析模型,创建算子的部分。
我还看到有op的registry,但是也没有找到registry怎么被调用的。
解答:
可以看一下CompileGraph函数。
register其实就是宏定义,利用的是全局变量初始化的时候调用构造函数加到一个map里。
https://gitee.com/mindspore/mindspore/blob/master/mindspore/lite/src/kernel_registry.h