Added bayesian_dense operator to relay (edc852d7) · Commits · Falk Selker / Tvm Likwid Unity · GitLab

Skip to content

GitLab

Explore

Sign in

Primary navigation

Project

T

Tvm Likwid Unity
- Activity
- Members
- Labels
- Issues
- Issue boards
- Milestones
- Wiki
- Environments
- Terraform modules
- Incidents

Snippets Groups Projects

edc852d7

Commit edc852d7 authored 1 year ago by Falk Selker

Downloads
- Patches
- Plain Diff

Added bayesian_dense operator to relay

parent bcf7abba

Loading

Changes 12

Hide whitespace changes

Inline Side-by-side

Showing

include/tvm/relay/attrs/nn.h 42 additions, 0 deletions

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

python/tvm/relay/op/nn/_nn.py
python/tvm/relay/op/nn/nn.py 77 additions, 0 deletions

python/tvm/relay/op/nn/nn.py
python/tvm/relay/op/strategy/generic.py 70 additions, 0 deletions

python/tvm/relay/op/strategy/generic.py
python/tvm/relay/op/strategy/x86.py 51 additions, 0 deletions

python/tvm/relay/op/strategy/x86.py
python/tvm/topi/generic/nn.py 37 additions, 0 deletions

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

python/tvm/topi/nn/__init__.py
python/tvm/topi/nn/bayesian_dense.py 253 additions, 0 deletions

python/tvm/topi/nn/bayesian_dense.py
python/tvm/topi/x86/__init__.py 1 addition, 0 deletions

python/tvm/topi/x86/__init__.py
python/tvm/topi/x86/bayesian_dense.py 173 additions, 0 deletions

python/tvm/topi/x86/bayesian_dense.py
src/relay/op/nn/nn.cc 76 additions, 0 deletions

src/relay/op/nn/nn.cc
src/relay/op/nn/nn.h 81 additions, 0 deletions

src/relay/op/nn/nn.h

with 915 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