TVM中tensorflow pb格式模型加载过程学习

1、通过tf将pb模型文件加载后生成GraphDef

这里需要注意，目前tvm只支持加载forzon的PB模型。

with tf.gfile.FastGFile(FLAGS.frozen_model_path, 'rb') as f:
        graph_def = tf.compat.v1.GraphDef()
        graph_def.ParseFromString(f.read())
        graph = tf.import_graph_def(graph_def, name='')

突然想起来我现在分析的不是TVM的源码，可能有公司代码泄漏风险，后面直接分析TVM代码后再继续写吧

©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容