[RELAY][VM] Add shape_of instruction (#5855)
authorZhi <5145158+zhiics@users.noreply.github.com>
Sun, 28 Jun 2020 17:05:50 +0000 (10:05 -0700)
committerGitHub <noreply@github.com>
Sun, 28 Jun 2020 17:05:50 +0000 (10:05 -0700)
commit731293462cd54b83566dfabf35daa36a75d56839
tree1cf183d8f26d5663f7b1fefcb4925d9a1c250bbb
parent17bd06a6142f21b553fd24bfaa9bdb4cc0ac8d13
[RELAY][VM] Add shape_of instruction (#5855)
15 files changed:
include/tvm/runtime/vm.h
python/tvm/relay/op/__init__.py
python/tvm/relay/op/vm/__init__.py [new file with mode: 0644]
python/tvm/relay/op/vm/_ffi_api.py [new file with mode: 0644]
python/tvm/relay/op/vm/vm.py [new file with mode: 0644]
python/tvm/relay/transform/memory_alloc.py
src/relay/backend/vm/compiler.cc
src/relay/op/tensor/unary.cc
src/relay/op/type_relations.cc
src/relay/op/type_relations.h
src/relay/op/vm/vm.cc [new file with mode: 0644]
src/relay/transforms/fold_constant.cc
src/runtime/vm/executable.cc
src/runtime/vm/vm.cc
tests/python/relay/test_vm_serialization.py