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
55592ece

Commit 55592ece authored 7 years ago by Tianqi Chen

Downloads
- Patches
- Plain Diff

[TOP] complete level2 (#8)

* [TOP] complete level2

* [TOP] add split

parent 13388655

No related branches found

No related tags found

No related merge requests found

Changes 9

Hide whitespace changes

Inline Side-by-side

Showing

nnvm/include/nnvm/top/nn.h 22 additions, 13 deletions

nnvm/include/nnvm/top/nn.h
nnvm/include/nnvm/top/tensor.h 17 additions, 2 deletions

nnvm/include/nnvm/top/tensor.h
nnvm/src/top/nn/convolution.cc 214 additions, 0 deletions

nnvm/src/top/nn/convolution.cc
nnvm/src/top/nn/nn.cc 5 additions, 223 deletions

nnvm/src/top/nn/nn.cc
nnvm/src/top/nn/nn_common.h 129 additions, 0 deletions

nnvm/src/top/nn/nn_common.h
nnvm/src/top/nn/pooling.cc 160 additions, 0 deletions

nnvm/src/top/nn/pooling.cc
nnvm/src/top/tensor/transform.cc 76 additions, 9 deletions

nnvm/src/top/tensor/transform.cc
nnvm/tests/python/unittest/test_infer_shape.py 126 additions, 0 deletions

nnvm/tests/python/unittest/test_infer_shape.py
nnvm/tests/python/unittest/test_top_level1.py 7 additions, 2 deletions

nnvm/tests/python/unittest/test_top_level1.py

with 756 additions and 249 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