projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: d9450f8) | patch
[Torch] Add cast to double, fix flatten conversion (#6357)
authormasahi <masahi129@gmail.com>
Sat, 29 Aug 2020 06:14:02 +0000 (15:14 +0900)
committerGitHub <noreply@github.com>
Sat, 29 Aug 2020 06:14:02 +0000 (11:44 +0530)
commit2d752d25bccd58f1c48b795fd275fde3a10c015d
tree9e63c64105277711cbf64c6a3963185477057c41tree | snapshot
parentd9450f8c9d8338aadf2632bad3aedb0b9416a8aecommit | diff
[Torch] Add cast to double, fix flatten conversion (#6357)

* support cast to double and fix flatten conversion

* also support batch flatten, add test

* add flatten test

* clean up
python/tvm/relay/frontend/pytorch.py diff | blob | history
tests/python/frontend/pytorch/test_forward.py diff | blob | history
Domain: Machine Learning / ML Framework;