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
44d8203f

Commit 44d8203f authored 6 years ago by nhynes Committed by Tianqi Chen 6 years ago

Downloads
- Patches
- Plain Diff

[NNVM][TOPI] Add gradients for broadcast_* ops (#1234)

parent 3f7cce3b

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/tuple.h 8 additions, 1 deletion

nnvm/include/nnvm/tuple.h
nnvm/python/nnvm/top/reduction.py 4 additions, 0 deletions

nnvm/python/nnvm/top/reduction.py
nnvm/src/top/tensor/broadcast.cc 57 additions, 4 deletions

nnvm/src/top/tensor/broadcast.cc
nnvm/src/top/tensor/reduce.cc 86 additions, 58 deletions

nnvm/src/top/tensor/reduce.cc
nnvm/src/top/tensor/transform.cc 1 addition, 3 deletions

nnvm/src/top/tensor/transform.cc
nnvm/tests/python/compiler/test_top_level4.py 77 additions, 2 deletions

nnvm/tests/python/compiler/test_top_level4.py
topi/include/topi/reduction.h 80 additions, 29 deletions

topi/include/topi/reduction.h
topi/python/topi/cuda/reduction.py 4 additions, 1 deletion

topi/python/topi/cuda/reduction.py
topi/tests/python_cpp/test_topi_reduce.py 1 addition, 0 deletions

topi/tests/python_cpp/test_topi_reduce.py

with 318 additions and 98 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