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
69d5fcab

Commit 69d5fcab authored 6 years ago by Tatsuya Nishiyama Committed by Tianqi Chen 6 years ago

Downloads
- Patches
- Plain Diff

Add count_include_pad support (#498)

* update tvm submodule

* Add count_include_pad support to avg_pool

parent fd6ad274

No related branches found

No related tags found

Loading

Changes 4

Hide whitespace changes

Inline Side-by-side

Showing

nnvm/include/nnvm/top/nn.h 31 additions, 2 deletions

nnvm/include/nnvm/top/nn.h
nnvm/src/compiler/fold_scale_axis.cc 8 additions, 8 deletions

nnvm/src/compiler/fold_scale_axis.cc
nnvm/src/top/nn/pooling.cc 22 additions, 18 deletions

nnvm/src/top/nn/pooling.cc
nnvm/tests/python/compiler/test_top_level2.py 36 additions, 0 deletions

nnvm/tests/python/compiler/test_top_level2.py

with 97 additions and 28 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