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
4001569e

Commit 4001569e authored 6 years ago by Siju Committed by Tianqi Chen 6 years ago

Downloads
- Patches
- Plain Diff

[RELAY][OP]Reduction operator framework, argmax, argmin (#1865)

parent fd392677

No related branches found

No related tags found

No related merge requests found

Changes 8

Hide whitespace changes

Inline Side-by-side

Showing

docs/langref/relay_op.rst 4 additions, 0 deletions

docs/langref/relay_op.rst
include/tvm/relay/type.h 7 additions, 0 deletions

include/tvm/relay/type.h
python/tvm/relay/__init__.py 1 addition, 0 deletions

python/tvm/relay/__init__.py
python/tvm/relay/op/__init__.py 1 addition, 0 deletions

python/tvm/relay/op/__init__.py
python/tvm/relay/op/reduce.py 64 additions, 0 deletions

python/tvm/relay/op/reduce.py
src/relay/op/tensor/reduce.cc 217 additions, 0 deletions

src/relay/op/tensor/reduce.cc
src/relay/pass/type_solver.cc 7 additions, 0 deletions

src/relay/pass/type_solver.cc
tests/python/relay/test_op_level4.py 90 additions, 0 deletions

tests/python/relay/test_op_level4.py

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