[TOPI] Fix atlest1d for reduce and squeeze (#2147)
Showing
- nnvm/include/nnvm/compiler/util.h 11 additions, 0 deletionsnnvm/include/nnvm/compiler/util.h
- nnvm/src/top/tensor/reduce.cc 15 additions, 19 deletionsnnvm/src/top/tensor/reduce.cc
- nnvm/src/top/tensor/transform.cc 2 additions, 2 deletionsnnvm/src/top/tensor/transform.cc
- topi/include/topi/detail/fuse.h 4 additions, 10 deletionstopi/include/topi/detail/fuse.h
- topi/include/topi/nn/l2_normalize.h 1 addition, 1 deletiontopi/include/topi/nn/l2_normalize.h
- topi/include/topi/nn/softmax.h 1 addition, 1 deletiontopi/include/topi/nn/softmax.h
- topi/include/topi/reduction.h 66 additions, 43 deletionstopi/include/topi/reduction.h
- topi/include/topi/transform.h 13 additions, 9 deletionstopi/include/topi/transform.h
- topi/python/topi/cuda/reduction.py 4 additions, 2 deletionstopi/python/topi/cuda/reduction.py
- topi/src/topi.cc 2 additions, 2 deletionstopi/src/topi.cc
- topi/tests/python/test_topi_reduce.py 4 additions, 0 deletionstopi/tests/python/test_topi_reduce.py
- topi/tests/python/test_topi_transform.py 1 addition, 4 deletionstopi/tests/python/test_topi_transform.py
- topi/tests/python_cpp/test_topi_transform.py 1 addition, 4 deletionstopi/tests/python_cpp/test_topi_transform.py
Loading
Please register or sign in to comment