[Relay][Op] Add compute, schedule, and tests for expand_dims and squeeze (#2133) (9473dca2) · 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
9473dca2

Commit 9473dca2 authored 6 years ago by Steven S. Lyubomirsky Committed by Tianqi Chen 6 years ago

Downloads
- Patches
- Plain Diff

[Relay][Op] Add compute, schedule, and tests for expand_dims and squeeze (#2133)

parent b1cf70a8

No related branches found

No related tags found

No related merge requests found

Changes 3

Hide whitespace changes

Inline Side-by-side

Showing

python/tvm/relay/op/_transform.py 44 additions, 1 deletion

python/tvm/relay/op/_transform.py
tests/python/relay/test_op_level1.py 17 additions, 0 deletions

tests/python/relay/test_op_level1.py
tests/python/relay/test_op_level3.py 17 additions, 0 deletions

tests/python/relay/test_op_level3.py

with 78 additions and 1 deletion

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