projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: bbcecc9) | patch
[Torch] Support logsumexp, clean up unnecessary infer_shape usage (#6374)
authormasahi <masahi129@gmail.com>
Wed, 2 Sep 2020 15:55:38 +0000 (00:55 +0900)
committerGitHub <noreply@github.com>
Wed, 2 Sep 2020 15:55:38 +0000 (21:25 +0530)
commitc0cbc11bfe14bc272f9782f3b67d157b411e194b
treed731689c98a3903e1605b3f1a3b7c5350eba79b4tree | snapshot
parentbbcecc951264f7c330a5c1797b1d90e5aed68871commit | diff
[Torch] Support logsumexp, clean up unnecessary infer_shape usage (#6374)

* clean up infer_shape usage, add logsumexp op

* add more tests for logsumexp

* remove commented code
python/tvm/relay/frontend/pytorch.py diff | blob | history
python/tvm/relay/frontend/qnn_torch.py diff | blob | history
tests/python/frontend/pytorch/test_forward.py diff | blob | history
Domain: Machine Learning / ML Framework;