[NNVM/TOPI][OP] Split : default axis to 0 and allow negative values - nump… (#1883) (6514849f) · 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
6514849f

Commit 6514849f authored 6 years ago by Siva Committed by Tianqi Chen 6 years ago

Downloads
- Patches
- Plain Diff

[NNVM/TOPI][OP] Split : default axis to 0 and allow negative values - nump… (#1883)

parent 7631873b

No related branches found

No related tags found

Loading

Changes 5

Hide whitespace changes

Inline Side-by-side

Showing

nnvm/include/nnvm/top/tensor.h 1 addition, 1 deletion

nnvm/include/nnvm/top/tensor.h
nnvm/src/top/tensor/transform.cc 18 additions, 9 deletions

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

nnvm/tests/python/unittest/test_infer_shape.py
topi/python/topi/transform.py 9 additions, 105 deletions

topi/python/topi/transform.py
topi/tests/python/test_topi_transform.py 1 addition, 1 deletion

topi/tests/python/test_topi_transform.py

with 33 additions and 116 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