如何从基础了解深度学习编译器中的数据流与控制流?
- 内容介绍
- 文章标签
- 相关推荐
本文共计2773个文字,预计阅读时间需要12分钟。
【GiantPandaCV导语】本文作为深度学习编译器的入门篇,介绍了深度学习框架的Data Flow和Control Flow,并基于TensorFlow解释了TensorFlow如何在静态图中实现Control Flow。
本文作为从零开始学深度学习编译器的番外篇,介绍了一下深度学习框架的Data Flow和Control Flow,并基于TensorFlow解释了TensorFlow是如何在静态图中实现Control Flow的。而对于动态图来说,是支持在Python层直接写Control Flow的,最后基于Pytorch介绍了如何将Python层的Control Flow导出到TorchScript模型以及ONNX模型。
0x0. 前言
本来是想在讲TVM Relay的时候提一下DataFlow和ControlFlow的,但是担心读者看到解析代码的文章打开就关了,所以这里用一篇简短的文章来介绍一下深度学习框架中的DataFlow和ControlFlow。
0x1. DataFlow
我记得我接触的第一个深度学习框架是TensorFlow1.x,本科毕业设计也是基于TensorFlow完成的,因此这里我将以TensorFlow1.x为例介绍一下DataFlow。
本文共计2773个文字,预计阅读时间需要12分钟。
【GiantPandaCV导语】本文作为深度学习编译器的入门篇,介绍了深度学习框架的Data Flow和Control Flow,并基于TensorFlow解释了TensorFlow如何在静态图中实现Control Flow。
本文作为从零开始学深度学习编译器的番外篇,介绍了一下深度学习框架的Data Flow和Control Flow,并基于TensorFlow解释了TensorFlow是如何在静态图中实现Control Flow的。而对于动态图来说,是支持在Python层直接写Control Flow的,最后基于Pytorch介绍了如何将Python层的Control Flow导出到TorchScript模型以及ONNX模型。
0x0. 前言
本来是想在讲TVM Relay的时候提一下DataFlow和ControlFlow的,但是担心读者看到解析代码的文章打开就关了,所以这里用一篇简短的文章来介绍一下深度学习框架中的DataFlow和ControlFlow。
0x1. DataFlow
我记得我接触的第一个深度学习框架是TensorFlow1.x,本科毕业设计也是基于TensorFlow完成的,因此这里我将以TensorFlow1.x为例介绍一下DataFlow。

