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
0e8d8fdd

Unverified Commit 0e8d8fdd authored 6 years ago by Haichen Shen Committed by GitHub 6 years ago

Downloads
- Patches
- Plain Diff

[Relay][OP] Add cast op (#2319)

* Add cast op
* Rename dtype_cast to cast
* Add additional safety check for String2TVMType
* Add missing relay op docs

parent 67fe4db0

No related branches found

No related tags found

Loading

Changes 6

Hide whitespace changes

Inline Side-by-side

Showing

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

docs/langref/relay_op.rst
include/tvm/runtime/packed_func.h 3 additions, 1 deletion

include/tvm/runtime/packed_func.h
python/tvm/relay/expr.py 1 addition, 1 deletion

python/tvm/relay/expr.py
python/tvm/relay/op/transform.py 20 additions, 0 deletions

python/tvm/relay/op/transform.py
src/relay/op/tensor/transform.cc 1 addition, 1 deletion

src/relay/op/tensor/transform.cc
tests/python/relay/test_op_level3.py 5 additions, 0 deletions

tests/python/relay/test_op_level3.py

with 37 additions and 3 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