Fix bugs with C++ TOPI flatten and relu (#869)
* Fix bugs with C++ TOPI flatten and relu * Added regression tests. Fixed typo in CMakeLists.txt. Fixed topi cpp import removed.
Showing
- CMakeLists.txt 1 addition, 1 deletionCMakeLists.txt
- topi/include/topi/nn.h 4 additions, 1 deletiontopi/include/topi/nn.h
- topi/include/topi/nn/flatten.h 1 addition, 1 deletiontopi/include/topi/nn/flatten.h
- topi/python/topi/__init__.py 1 addition, 0 deletionstopi/python/topi/__init__.py
- topi/tests/python_cpp/test_topi_basic.py 5 additions, 0 deletionstopi/tests/python_cpp/test_topi_basic.py
- topi/tests/python_cpp/test_topi_relu.py 5 additions, 3 deletionstopi/tests/python_cpp/test_topi_relu.py
Loading
Please register or sign in to comment