projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 9a42cb6) | patch
[Relay] Enhance relay.split(), allow splitted dim to be dynamic (#6289)
authorlixiaoquan <radioheads@163.com>
Wed, 2 Sep 2020 06:04:37 +0000 (14:04 +0800)
committerGitHub <noreply@github.com>
Wed, 2 Sep 2020 06:04:37 +0000 (23:04 -0700)
commit6ed242eadebee61e7da2ad468519661acfe31b84
treef837a59ee581d098923208fa15d9b118d781e432tree | snapshot
parent9a42cb63d29710a37fc34211d0cdc502d0626771commit | diff
[Relay] Enhance relay.split(), allow splitted dim to be dynamic (#6289)

* [Relay] Enhance relay.split(), allow splitted dim to be dynamic

* Add assert in shape function

* Fix CI
include/tvm/topi/transform.h diff | blob | history
python/tvm/relay/op/_transform.py diff | blob | history
src/relay/op/tensor/transform.cc diff | blob | history
tests/python/relay/test_any.py diff | blob | history
Domain: Machine Learning / ML Framework;