TVM编译Tensorflow模型

最新推荐文章于 2022-04-25 13:29:17 发布

yhwang-hub

最新推荐文章于 2022-04-25 13:29:17 发布

阅读量502

点赞数 1

分类专栏： TVM

本文链接：https://blog.csdn.net/qq_33287871/article/details/116040908

版权

TVM 专栏收录该内容

41 篇文章 31 订阅 ¥19.90 ¥99.00

订阅专栏

超级会员免费看

本文提供了一个入门教程，介绍如何利用TVM编译和执行TensorFlow模型。首先，安装TensorFlow Python模块，然后从protobuf文件导入模型，解码图像，并将TensorFlow图转换为Relay表达式。接着，编译Relay图并生成可在TVM上运行的库。最后，执行模型并处理输出，使其适应目标平台。

摘要由CSDN通过智能技术生成

编译Tensorflow模型

讲解
下载所需文件
导入模型
解码图像
将图形导入到中继
构建Relay
在TVM上执行可移植图形
处理输出
关于张量流的推理

本文是使用TVM部署Tensorflow模型的入门教程。
首先，需要安装tensorflow python模块。
请参考https://www.tensorflow.org/install

# tvm,relay
import tvm
from tvm import relay
from tvm import te

#os and numpy
import numpy as np
import os.path

# Tensorflow imports
import tensorflow as tf

try:
    tf_compat_v1=tf.compat.v1
except ImportError:
    tf_compat_v1=tf

# Tensorflow utility function
import tvm.relay.testing.tf as tf_testing

# Base location for model related files.
repo_base="https://github.com/dmlc/web-data/raw/main/tensorflow/models/InceptionV1/"

# Test image
img_name="elephant-299.jpg"
image_url=os.path.join(repo_base,img_name)

了解本专栏