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
07e56b9a

Commit 07e56b9a authored 7 years ago by Yuwei HU Committed by Tianqi Chen 7 years ago

Downloads
- Patches
- Plain Diff

update depthwise_conv2d schedule and testing (#328)

parent 8edd047b

No related branches found

No related tags found

No related merge requests found

Changes 7

Hide whitespace changes

Inline Side-by-side

Showing

topi/python/topi/cuda/__init__.py 1 addition, 1 deletion

topi/python/topi/cuda/__init__.py
topi/python/topi/cuda/depthwise_conv2d.py 29 additions, 30 deletions

topi/python/topi/cuda/depthwise_conv2d.py
topi/python/topi/nn/mapping.py 1 addition, 1 deletion

topi/python/topi/nn/mapping.py
topi/python/topi/testing/__init__.py 1 addition, 0 deletions

topi/python/topi/testing/__init__.py
topi/python/topi/testing/depthwise_conv2d_python.py 62 additions, 0 deletions

topi/python/topi/testing/depthwise_conv2d_python.py
topi/recipe/conv/depthwise_conv2d_test.py 34 additions, 59 deletions

topi/recipe/conv/depthwise_conv2d_test.py
topi/tests/python/test_topi_depthwise_conv2d.py 86 additions, 0 deletions

topi/tests/python/test_topi_depthwise_conv2d.py

with 214 additions and 91 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