[TOPI] update c++ pool and softmax (#905) (6d4eb2e8) · 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
6d4eb2e8

Commit 6d4eb2e8 authored 7 years ago by masahi Committed by Tianqi Chen 7 years ago

Downloads
- Patches
- Plain Diff

[TOPI] update c++ pool and softmax (#905)

* update c++ pool and softmax

* clean up reduce axis

parent 99c96962

No related branches found

No related tags found

No related merge requests found

Changes 5

Hide whitespace changes

Inline Side-by-side

Showing

topi/include/topi/nn/pooling.h 124 additions, 6 deletions

topi/include/topi/nn/pooling.h
topi/include/topi/nn/softmax.h 54 additions, 17 deletions

topi/include/topi/nn/softmax.h
topi/src/topi.cc 2 additions, 2 deletions

topi/src/topi.cc
topi/tests/python_cpp/test_topi_pooling.py 1 addition, 1 deletion

topi/tests/python_cpp/test_topi_pooling.py
topi/tests/python_cpp/test_topi_softmax.py 1 addition, 1 deletion

topi/tests/python_cpp/test_topi_softmax.py

with 182 additions and 27 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