projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 7a41971) | patch
support aten::type_as in the pytorch frontend (#5787)
authorRand Xie <randxiexyy29@gmail.com>
Sat, 13 Jun 2020 04:52:45 +0000 (21:52 -0700)
committerGitHub <noreply@github.com>
Sat, 13 Jun 2020 04:52:45 +0000 (13:52 +0900)
commit59f5cbe921cf329febcd9d6eff2df94d80f1c523
tree50b168db3822888c56065b3786c046ff2d975484tree | snapshot
parent7a419718c121164fc260864014e1d0d81f556949commit | diff
support aten::type_as in the pytorch frontend (#5787)

* support aten::type_as in the pytorch frontend

* use _convert_data_type to convert torch type to tvm type and add more types in the type_as test
python/tvm/relay/frontend/pytorch.py diff | blob | history
tests/python/frontend/pytorch/test_forward.py diff | blob | history
Domain: Machine Learning / ML Framework;