Update Graph Support for Batching, Fix Swapping (#37) (8c9758b6) · 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
8c9758b6

Commit 8c9758b6 authored 6 years ago by Tianqi Chen

Downloads
- Patches
- Plain Diff

Update Graph Support for Batching, Fix Swapping (#37)

* fix graph transform for batch dimension

* fix

* fix

parent a96a4a9b

No related branches found

No related tags found

No related merge requests found

Changes 4

Hide whitespace changes

Inline Side-by-side

Showing

vta/examples/resnet18/pynq/imagenet_predict.py 19 additions, 6 deletions

vta/examples/resnet18/pynq/imagenet_predict.py
vta/python/vta/graph.py 44 additions, 32 deletions

vta/python/vta/graph.py
vta/src/runtime.cc 3 additions, 1 deletion

vta/src/runtime.cc
vta/src/sim/sim_driver.cc 1 addition, 0 deletions

vta/src/sim/sim_driver.cc

with 67 additions and 39 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