-
- Downloads
[GRADIENT] Add backward operator to enable backward graph (#276)
* Update docs * Add backward operator to enable backward graph * Fix testing * Refactor top level1 test code * Fix format * Test * Added zeros ones op * Register fill_like operator * Fix unit test
Showing
- nnvm/docs/top.rst 4 additions, 0 deletionsnnvm/docs/top.rst
- nnvm/include/nnvm/top/tensor.h 28 additions, 12 deletionsnnvm/include/nnvm/top/tensor.h
- nnvm/python/nnvm/top/registry.py 1 addition, 1 deletionnnvm/python/nnvm/top/registry.py
- nnvm/src/top/elemwise_op_common.h 10 additions, 0 deletionsnnvm/src/top/elemwise_op_common.h
- nnvm/src/top/op_common.h 46 additions, 1 deletionnnvm/src/top/op_common.h
- nnvm/src/top/tensor/elemwise.cc 257 additions, 19 deletionsnnvm/src/top/tensor/elemwise.cc
- nnvm/tests/python/compiler/test_top_level1.py 171 additions, 124 deletionsnnvm/tests/python/compiler/test_top_level1.py
Loading
Please register or sign in to comment