projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: cef9609) | patch
Feat(frontend-pytorch): Add input types argument and Support cast to float16. (#6546)
authorLiu Qi <nolan.liou@gmail.com>
Fri, 25 Sep 2020 14:04:30 +0000 (22:04 +0800)
committerGitHub <noreply@github.com>
Fri, 25 Sep 2020 14:04:30 +0000 (23:04 +0900)
commit22e159fa3fe9d38bb1d0ce69c3788eef1946e341
tree00e9217535fb029c4061cdc07a380c42313bb649tree | snapshot
parentcef9609e188ec6818d594c153d6cc2861abb724ecommit | diff
Feat(frontend-pytorch): Add input types argument and Support cast to float16. (#6546)

1. Add input types argument for converting TorchScript file.
2. Support casting float32 to float16 when converting to operation.
python/tvm/relay/frontend/pytorch.py diff | blob | history
tests/python/frontend/pytorch/test_forward.py diff | blob | history
Domain: Machine Learning / ML Framework;