projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 91ea9bc) | patch
[Torch] Fix dtype handling for modules with integer parameters (#6311)
authormasahi <masahi129@gmail.com>
Fri, 21 Aug 2020 02:29:45 +0000 (11:29 +0900)
committerGitHub <noreply@github.com>
Fri, 21 Aug 2020 02:29:45 +0000 (07:59 +0530)
commit470dfc30a203bdd0bd05eb4cd7034dda9e85e617
treed297931ad894d929b8acbcc9dbf32be962482750tree | snapshot
parent91ea9bce8bffda702698cbceb8d66b529cbb8812commit | diff
[Torch] Fix dtype handling for modules with integer parameters (#6311)

* return the correct type for GetAttr node

* keep _get_pytorch_value_type intact

* add test and handle quantized param
python/tvm/relay/frontend/pytorch.py diff | blob | history
tests/python/frontend/pytorch/test_forward.py diff | blob | history
Domain: Machine Learning / ML Framework;