[RELAY]sch & comp for ops in nn.py (#2092) (6edb3564) · 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
6edb3564

Commit 6edb3564 authored 6 years ago by Siju Committed by Tianqi Chen 6 years ago

Downloads
- Patches
- Plain Diff

[RELAY]sch & comp for ops in nn.py (#2092)

parent 7761416f

No related branches found

No related tags found

Loading

Changes 5

Hide whitespace changes

Inline Side-by-side

Showing

include/tvm/relay/attrs/nn.h 1 addition, 1 deletion

include/tvm/relay/attrs/nn.h
python/tvm/relay/op/nn/_nn.py 45 additions, 0 deletions

python/tvm/relay/op/nn/_nn.py
src/relay/op/nn/nn.cc 28 additions, 4 deletions

src/relay/op/nn/nn.cc
tests/python/relay/test_op_level2.py 62 additions, 0 deletions

tests/python/relay/test_op_level2.py
tests/python/relay/test_op_level3.py 41 additions, 0 deletions

tests/python/relay/test_op_level3.py

with 177 additions and 5 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