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
a53d8d01

Commit a53d8d01 authored 7 years ago by Tianqi Chen

Downloads
- Patches
- Plain Diff

[PASS] Enhance scale fold axis (#424)

parent 89c124bc

No related branches found

No related tags found

Loading

Changes 3

Hide whitespace changes

Inline Side-by-side

Showing

nnvm/src/compiler/fold_scale_axis.cc 327 additions, 86 deletions

nnvm/src/compiler/fold_scale_axis.cc
nnvm/src/pass/plan_memory.cc 1 addition, 1 deletion

nnvm/src/pass/plan_memory.cc
nnvm/tests/python/compiler/test_fold_axis.py 70 additions, 9 deletions

nnvm/tests/python/compiler/test_fold_axis.py

with 398 additions and 96 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