[NNVM] Initial mixed precision support of conv2d (#1356) (b154e6b9) · 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
b154e6b9

Commit b154e6b9 authored 6 years ago by Tianqi Chen Committed by GitHub 6 years ago

Downloads
- Patches
- Plain Diff

[NNVM] Initial mixed precision support of conv2d (#1356)

parent 4fb58115

No related branches found

No related tags found

Loading

Changes 6

Hide whitespace changes

Inline Side-by-side

Showing

nnvm/include/nnvm/top/nn.h 7 additions, 0 deletions

nnvm/include/nnvm/top/nn.h
nnvm/python/nnvm/top/nn.py 9 additions, 3 deletions

nnvm/python/nnvm/top/nn.py
nnvm/src/top/nn/convolution.cc 27 additions, 3 deletions

nnvm/src/top/nn/convolution.cc
nnvm/tests/python/compiler/test_top_level2.py 32 additions, 2 deletions

nnvm/tests/python/compiler/test_top_level2.py
nnvm/tests/python/unittest/test_infer_shape.py 22 additions, 0 deletions

nnvm/tests/python/unittest/test_infer_shape.py
topi/python/topi/nn/depthwise_conv2d.py 13 additions, 4 deletions

topi/python/topi/nn/depthwise_conv2d.py

with 110 additions and 12 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