Nnvm 源码阅读

未完成。

NNVM is not a deep learning library. It is a modular, decentralized and lightweight part to help build deep learning libraries.

NNVM offers one such part, it provides a generic way to do computation graph optimization such as memory reduction, device allocation and more while being agnostic to the operator interface definition and how operators are executed. NNVM is inspired by LLVM, aiming to be a high level intermediate representation library for neural nets and computation graphs generation and optimizations.

• .h/头文件 + .cc源码
  • include/dmlc/memory.h
    • src/pass/plan_memory.cc
  • include/dmlc/graph.h
  • include/dmlc/node.h

1. memory.h and plan_memory.cc

2. nnvm/include/nnvm/graph.h

3. nnvm/include/nnvm/node.h