[DSL/TE] Scalar support for `te.extern` (#6079)
authorHaibin Lin <linhaibin.eric@gmail.com>
Tue, 21 Jul 2020 00:49:59 +0000 (17:49 -0700)
committerGitHub <noreply@github.com>
Tue, 21 Jul 2020 00:49:59 +0000 (17:49 -0700)
commit7ac78d96611ab3a069fae9eb5210e433eb240cc5
treedb43589aa14a8b9e9af0bd479c328a3542ac565f
parentb95ac1fb6bdc19b547a49e4a5f13432e57dc4d2c
[DSL/TE] Scalar support for `te.extern`  (#6079)

* fix make shape with scalar shapes

* add test

* add test

* remove scalar shape assertion

* fix the data type for overflow problems

* add extra tests

Co-authored-by: Ubuntu <ubuntu@ip-172-31-42-138.ec2.internal>
python/tvm/te/operation.py
python/tvm/tir/op.py
src/tir/transforms/lower_tvm_builtin.cc
tests/python/unittest/test_te_tensor.py