-
- Downloads
[NNVM][TOPI] Add mean and product operators (#1628)
* Add mean and product operators * Fix typo * Fix lint * fix test * Fix gpu schedule * Update doc * remove mean from topi * Add nnvm test * Fix cuda schedule * Remove cuda schedule
Showing
- docs/api/python/topi.rst 2 additions, 0 deletionsdocs/api/python/topi.rst
- docs/nnvm_top.rst 4 additions, 0 deletionsdocs/nnvm_top.rst
- include/tvm/expr.h 1 addition, 0 deletionsinclude/tvm/expr.h
- include/tvm/ir_operator.h 6 additions, 0 deletionsinclude/tvm/ir_operator.h
- nnvm/python/nnvm/top/reduction.py 8 additions, 0 deletionsnnvm/python/nnvm/top/reduction.py
- nnvm/src/top/tensor/reduce.cc 64 additions, 0 deletionsnnvm/src/top/tensor/reduce.cc
- nnvm/tests/python/compiler/test_top_level4.py 10 additions, 0 deletionsnnvm/tests/python/compiler/test_top_level4.py
- src/lang/ir_operator.cc 9 additions, 0 deletionssrc/lang/ir_operator.cc
- topi/include/topi/reduction.h 22 additions, 0 deletionstopi/include/topi/reduction.h
- topi/python/topi/reduction.py 32 additions, 132 deletionstopi/python/topi/reduction.py
- topi/src/topi.cc 5 additions, 0 deletionstopi/src/topi.cc
- topi/tests/python/test_topi_reduce.py 1 addition, 0 deletionstopi/tests/python/test_topi_reduce.py
- topi/tests/python_cpp/test_topi_reduce.py 25 additions, 13 deletionstopi/tests/python_cpp/test_topi_reduce.py
Loading
Please register or sign in to comment