[TOPI][DARKNET]Yolo op added (#1372)
Showing
- nnvm/python/nnvm/frontend/darknet.py 2 additions, 2 deletionsnnvm/python/nnvm/frontend/darknet.py
- nnvm/python/nnvm/top/vision.py 7 additions, 7 deletionsnnvm/python/nnvm/top/vision.py
- nnvm/src/top/vision/yolo/region.cc 1 addition, 1 deletionnnvm/src/top/vision/yolo/region.cc
- nnvm/src/top/vision/yolo/region.h 3 additions, 3 deletionsnnvm/src/top/vision/yolo/region.h
- nnvm/src/top/vision/yolo/reorg.cc 1 addition, 1 deletionnnvm/src/top/vision/yolo/reorg.cc
- nnvm/src/top/vision/yolo/reorg.h 3 additions, 3 deletionsnnvm/src/top/vision/yolo/reorg.h
- topi/include/topi/vision/yolo/region.h 6 additions, 6 deletionstopi/include/topi/vision/yolo/region.h
- topi/include/topi/vision/yolo/yolo.h 58 additions, 0 deletionstopi/include/topi/vision/yolo/yolo.h
- topi/python/topi/cpp.py 2 additions, 2 deletionstopi/python/topi/cpp.py
- topi/python/topi/testing/__init__.py 1 addition, 0 deletionstopi/python/topi/testing/__init__.py
- topi/python/topi/testing/yolo_python.py 43 additions, 0 deletionstopi/python/topi/testing/yolo_python.py
- topi/python/topi/vision/__init__.py 1 addition, 1 deletiontopi/python/topi/vision/__init__.py
- topi/python/topi/vision/yolo/__init__.py 1 addition, 0 deletionstopi/python/topi/vision/yolo/__init__.py
- topi/python/topi/vision/yolo/region.py 1 addition, 1 deletiontopi/python/topi/vision/yolo/region.py
- topi/python/topi/vision/yolo/yolo.py 30 additions, 0 deletionstopi/python/topi/vision/yolo/yolo.py
- topi/src/topi.cc 9 additions, 3 deletionstopi/src/topi.cc
- topi/tests/python/test_topi_region.py 1 addition, 1 deletiontopi/tests/python/test_topi_region.py
- topi/tests/python_cpp/test_topi_region.py 1 addition, 1 deletiontopi/tests/python_cpp/test_topi_region.py
- topi/tests/python_cpp/test_topi_yolo.py 49 additions, 0 deletionstopi/tests/python_cpp/test_topi_yolo.py
Loading
Please register or sign in to comment