Skip to content

GitLab

Explore

Sign in

Primary navigation

Project

T

tvm
- Activity
- Members
- Labels
- Issues
- Issue boards
- Milestones
- Wiki
- Releases
- Model registry
- Environments
- Incidents
- Service Desk

Snippets Groups Projects

- cld
- ml
- tvm
- Commits
e6863990

Commit e6863990 authored 6 years ago by Leyuan Wang Committed by Tianqi Chen 6 years ago

Downloads
- Patches
- Plain Diff

[TOPI] Add GPU SSD (#1397)

parent 3a0b757c

No related branches found

No related tags found

No related merge requests found

Changes 8

Hide whitespace changes

Inline Side-by-side

Showing

topi/python/topi/cuda/__init__.py 4 additions, 2 deletions

topi/python/topi/cuda/__init__.py
topi/python/topi/cuda/nms.py 354 additions, 0 deletions

topi/python/topi/cuda/nms.py
topi/python/topi/cuda/ssd/__init__.py 5 additions, 0 deletions

topi/python/topi/cuda/ssd/__init__.py
topi/python/topi/cuda/ssd/multibox.py 360 additions, 0 deletions

topi/python/topi/cuda/ssd/multibox.py
topi/python/topi/cuda/vision.py 73 additions, 5 deletions

topi/python/topi/cuda/vision.py
topi/tests/python/test_topi_clip.py 9 additions, 5 deletions

topi/tests/python/test_topi_clip.py
topi/tests/python/test_topi_conv2d_nchw.py 18 additions, 1 deletion

topi/tests/python/test_topi_conv2d_nchw.py
topi/tests/python/test_topi_vision.py 17 additions, 8 deletions

topi/tests/python/test_topi_vision.py

with 840 additions and 21 deletions

Loading

0% Loading or .

You are about to add 0 people to the discussion. Proceed with caution.

Finish editing this message first!

Please register or sign in to comment