Extend TensorComputeOp to allow scalar inputs (#2606). (#3300)
authorJessica Davies <46725573+jdavies-huawei@users.noreply.github.com>
Sat, 22 Jun 2019 04:22:54 +0000 (00:22 -0400)
committerziheng <ziheng@apache.org>
Sat, 22 Jun 2019 04:22:54 +0000 (21:22 -0700)
commite9634eadddb177b56209aae1509599b478900d98
tree43dd8c6fa39682c5323d1e4c058c369e0224a766
parent1e66455b867e9f3e517d2edefc8bf70f78ab417a
Extend TensorComputeOp to allow scalar inputs (#2606). (#3300)
include/tvm/operation.h
include/tvm/tensor_intrin.h
python/tvm/api.py
python/tvm/tensor_intrin.py
src/lang/tensor.cc
src/op/tensor_compute_op.cc
src/schedule/schedule_dataflow_rewrite.cc
tests/python/unittest/test_lang_schedule.py