-
- Downloads
[GRADIENT] Register more gradient operators (#300)
* Add conv2d max_pool backward op * Added tests * Fix testing * Address comments * Change dot to matmul * Address comments * Break down indicator function * Make greater, less numpy compatible
Showing
- nnvm/docs/top.rst 12 additions, 0 deletionsnnvm/docs/top.rst
- nnvm/include/nnvm/top/tensor.h 74 additions, 4 deletionsnnvm/include/nnvm/top/tensor.h
- nnvm/python/nnvm/compiler/build_module.py 1 addition, 1 deletionnnvm/python/nnvm/compiler/build_module.py
- nnvm/python/nnvm/compiler/graph_util.py 57 additions, 0 deletionsnnvm/python/nnvm/compiler/graph_util.py
- nnvm/python/nnvm/graph.py 0 additions, 36 deletionsnnvm/python/nnvm/graph.py
- nnvm/src/pass/gradient.cc 7 additions, 2 deletionsnnvm/src/pass/gradient.cc
- nnvm/src/top/elemwise_op_common.h 58 additions, 5 deletionsnnvm/src/top/elemwise_op_common.h
- nnvm/src/top/nn/convolution.cc 36 additions, 1 deletionnnvm/src/top/nn/convolution.cc
- nnvm/src/top/nn/nn.cc 123 additions, 2 deletionsnnvm/src/top/nn/nn.cc
- nnvm/src/top/nn/pooling.cc 22 additions, 0 deletionsnnvm/src/top/nn/pooling.cc
- nnvm/src/top/op_common.h 52 additions, 3 deletionsnnvm/src/top/op_common.h
- nnvm/src/top/tensor/elemwise.cc 103 additions, 86 deletionsnnvm/src/top/tensor/elemwise.cc
- nnvm/src/top/tensor/matrix_op.cc 138 additions, 0 deletionsnnvm/src/top/tensor/matrix_op.cc
- nnvm/src/top/tensor/reduce.cc 69 additions, 14 deletionsnnvm/src/top/tensor/reduce.cc
- nnvm/src/top/tensor/transform.cc 124 additions, 5 deletionsnnvm/src/top/tensor/transform.cc
- nnvm/tests/cpp/tuple_test.cc 1 addition, 1 deletionnnvm/tests/cpp/tuple_test.cc
- nnvm/tests/python/unittest/test_graph.py 3 additions, 2 deletionsnnvm/tests/python/unittest/test_graph.py
- nnvm/tests/python/unittest/test_graph_gradient.py 136 additions, 0 deletionsnnvm/tests/python/unittest/test_graph_gradient.py
Loading
Please register or sign in to comment