tinyflow
xwzheng1020
这个作者很懒,什么都没留下…
展开
-
tinyflow源码阅读-0
似乎有必要介绍下NNVMNNVM不是一个深度学习库,而是一个去中心化的、轻量的深度学习系统构建辅助模块。 不同于很多深度学习系统提供端到端的解决方案,NNVM尝试为自定义一个深度学习系统提供可能和关键性模块。它提供一种通用的方式进行计算图优化,如内存节约和设备调度,而无关于运算接口定义和运算执行。它更像是神经网络、计算图生成和优化的中间表示层。 示例:NNVM_REGISTER_OP(add)原创 2017-06-14 11:37:47 · 636 阅读 · 0 评论 -
tinyflow源码阅读-0
tinyflow简介——从tqchen/tinyflow翻译而来TinyFlow是NNVM的示例代码,旨在展示如何用少量的代码(2K行)构建自己的深度学习系统。构建的DL系统基于图计算,具有TensorFlow式的API。其中,运算符由Torch7实现,而更侧重于展示系统概念。 TinyFlow可以运行在CPU和GPU上,代码结构如下: 927行运算符代码(C++中嵌入Lua) 734行运算执行代翻译 2017-06-13 15:49:27 · 2325 阅读 · 0 评论