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
8dd928c7

Commit 8dd928c7 authored 6 years ago by Wuwei Lin Committed by masahi 6 years ago

Downloads
- Patches
- Plain Diff

[TOPI] Add roi align (#2350)

* [TOPI] Add roi align

* Refactor bilinear in image resize

* Rename to roi_align_nchw

* Fix

parent 3708b311

No related branches found

No related tags found

No related merge requests found

Changes 10

Hide whitespace changes

Inline Side-by-side

Showing

topi/include/topi/image/resize.h 40 additions, 23 deletions

topi/include/topi/image/resize.h
topi/python/topi/cuda/vision.py 5 additions, 0 deletions

topi/python/topi/cuda/vision.py
topi/python/topi/generic/vision.py 17 additions, 0 deletions

topi/python/topi/generic/vision.py
topi/python/topi/testing/__init__.py 1 addition, 0 deletions

topi/python/topi/testing/__init__.py
topi/python/topi/testing/roi_align_python.py 63 additions, 0 deletions

topi/python/topi/testing/roi_align_python.py
topi/python/topi/vision/__init__.py 1 addition, 0 deletions

topi/python/topi/vision/__init__.py
topi/python/topi/vision/rcnn/__init__.py 3 additions, 0 deletions

topi/python/topi/vision/rcnn/__init__.py
topi/python/topi/vision/rcnn/roi_align.py 85 additions, 0 deletions

topi/python/topi/vision/rcnn/roi_align.py
topi/src/topi.cc 5 additions, 0 deletions

topi/src/topi.cc
topi/tests/python/test_topi_vision.py 54 additions, 1 deletion

topi/tests/python/test_topi_vision.py

with 274 additions and 24 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