enable AlterOpLayout to keep OP unchanged (#471) (7f5d22d7) · 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
7f5d22d7

Commit 7f5d22d7 authored 6 years ago by Yizhi Liu Committed by Tianqi Chen 6 years ago

Downloads
- Patches
- Plain Diff

enable AlterOpLayout to keep OP unchanged (#471)

parent be968fef

No related branches found

No related tags found

Loading

Changes 7

Hide whitespace changes

Inline Side-by-side

Showing

nnvm/include/nnvm/compiler/op_attr_types.h 6 additions, 3 deletions

nnvm/include/nnvm/compiler/op_attr_types.h
nnvm/python/nnvm/top/nn.py 4 additions, 0 deletions

nnvm/python/nnvm/top/nn.py
nnvm/src/compiler/alter_op_layout.cc 5 additions, 3 deletions

nnvm/src/compiler/alter_op_layout.cc
nnvm/src/compiler/fold_scale_axis.cc 3 additions, 3 deletions

nnvm/src/compiler/fold_scale_axis.cc
nnvm/src/compiler/packed_func_ext.cc 7 additions, 2 deletions

nnvm/src/compiler/packed_func_ext.cc
nnvm/src/top/nn/nn_common.h 1 addition, 1 deletion

nnvm/src/top/nn/nn_common.h
nnvm/tests/python/compiler/test_alter_op_layout.py 1 addition, 1 deletion

nnvm/tests/python/compiler/test_alter_op_layout.py

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