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
3f599a60

Commit 3f599a60 authored 7 years ago by Xingjian Shi Committed by Tianqi Chen 6 years ago

Downloads
- Patches
- Plain Diff

add squeeze (#52)

* add transform

* fix

* update doc

* Update tvm

parent 5541a275

No related branches found

No related tags found

No related merge requests found

Changes 8

Hide whitespace changes

Inline Side-by-side

Showing

nnvm/docs/top.rst 2 additions, 0 deletions

nnvm/docs/top.rst
nnvm/include/nnvm/top/tensor.h 10 additions, 0 deletions

nnvm/include/nnvm/top/tensor.h
nnvm/python/nnvm/compiler/compile_engine.py 1 addition, 1 deletion

nnvm/python/nnvm/compiler/compile_engine.py
nnvm/python/nnvm/top/transform.py 10 additions, 0 deletions

nnvm/python/nnvm/top/transform.py
nnvm/src/top/op_common.h 1 addition, 0 deletions

nnvm/src/top/op_common.h
nnvm/src/top/tensor/transform.cc 74 additions, 0 deletions

nnvm/src/top/tensor/transform.cc
nnvm/tests/python/compiler/test_top_level1.py 25 additions, 0 deletions

nnvm/tests/python/compiler/test_top_level1.py
nnvm/tests/python/compiler/test_top_level4.py 0 additions, 1 deletion

nnvm/tests/python/compiler/test_top_level4.py

with 123 additions and 2 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