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
669e9c15

Commit 669e9c15 authored 7 years ago by Yao Wang Committed by Tianqi Chen 6 years ago

Downloads
- Patches
- Plain Diff

Add gradient graph (#280)

* Add creating gradient symbol

* Fix lint

* Address comments

* Fix typo

* Address comment

parent 492a37c5

No related branches found

No related tags found

Loading

Changes 5

Hide whitespace changes

Inline Side-by-side

Showing

nnvm/docs/top.rst 12 additions, 4 deletions

nnvm/docs/top.rst
nnvm/python/nnvm/graph.py 36 additions, 0 deletions

nnvm/python/nnvm/graph.py
nnvm/src/pass/gradient.cc 2 additions, 2 deletions

nnvm/src/pass/gradient.cc
nnvm/src/top/tensor/elemwise.cc 111 additions, 9 deletions

nnvm/src/top/tensor/elemwise.cc
nnvm/tests/python/unittest/test_graph.py 18 additions, 0 deletions

nnvm/tests/python/unittest/test_graph.py

with 179 additions and 15 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