[TOPI] LRN & L2norm Operator (#1051) (ce02ee3b) · 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
ce02ee3b

Commit ce02ee3b authored 6 years ago by Pariksheet Pinjari Committed by Tianqi Chen 6 years ago

Downloads
- Patches
- Plain Diff

[TOPI] LRN & L2norm Operator (#1051)

parent b3f09b01

No related branches found

No related tags found

No related merge requests found

Changes 10

Hide whitespace changes

Inline Side-by-side

Showing

topi/python/topi/cuda/__init__.py 1 addition, 0 deletions

topi/python/topi/cuda/__init__.py
topi/python/topi/cuda/nn.py 91 additions, 0 deletions

topi/python/topi/cuda/nn.py
topi/python/topi/generic/nn.py 36 additions, 0 deletions

topi/python/topi/generic/nn.py
topi/python/topi/nn/__init__.py 2 additions, 0 deletions

topi/python/topi/nn/__init__.py
topi/python/topi/nn/l2_norm.py 35 additions, 0 deletions

topi/python/topi/nn/l2_norm.py
topi/python/topi/nn/local_response_norm.py 68 additions, 0 deletions

topi/python/topi/nn/local_response_norm.py
topi/python/topi/rocm/__init__.py 1 addition, 0 deletions

topi/python/topi/rocm/__init__.py
topi/python/topi/rocm/nn.py 13 additions, 0 deletions

topi/python/topi/rocm/nn.py
topi/tests/python/test_topi_l2norm.py 70 additions, 0 deletions

topi/tests/python/test_topi_l2norm.py
topi/tests/python/test_topi_lrn.py 95 additions, 0 deletions

topi/tests/python/test_topi_lrn.py

with 412 additions and 0 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