[TOPI][CUDA] int8 group conv2d (#2075) (5c410c4c) · Commits · cld / ml / tvm · GitLab

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
5c410c4c

Commit 5c410c4c authored 6 years ago by Wuwei Lin Committed by Tianqi Chen 6 years ago

Downloads
- Patches
- Plain Diff

[TOPI][CUDA] int8 group conv2d (#2075)

parent 3ee13fc5

No related branches found

No related tags found

No related merge requests found

Changes 10

Hide whitespace changes

Inline Side-by-side

Showing

nnvm/python/nnvm/top/nn.py 5 additions, 0 deletions

nnvm/python/nnvm/top/nn.py
python/tvm/autotvm/task/nnvm_integration.py 13 additions, 1 deletion

python/tvm/autotvm/task/nnvm_integration.py
topi/python/topi/cuda/__init__.py 2 additions, 1 deletion

topi/python/topi/cuda/__init__.py
topi/python/topi/cuda/group_conv2d_nchw.py 308 additions, 0 deletions

topi/python/topi/cuda/group_conv2d_nchw.py
topi/python/topi/generic/nn.py 19 additions, 0 deletions

topi/python/topi/generic/nn.py
topi/python/topi/nn/conv2d.py 77 additions, 0 deletions

topi/python/topi/nn/conv2d.py
topi/python/topi/testing/conv2d_nchw_python.py 35 additions, 2 deletions

topi/python/topi/testing/conv2d_nchw_python.py
topi/tests/python/common.py 15 additions, 0 deletions

topi/tests/python/common.py
topi/tests/python/test_topi_conv2d_int8.py 1 addition, 12 deletions

topi/tests/python/test_topi_conv2d_int8.py
topi/tests/python/test_topi_group_conv2d.py 215 additions, 0 deletions

topi/tests/python/test_topi_group_conv2d.py

with 690 additions and 16 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