projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: e5044cb) | patch
Fix stride default value None in torch.nn.functional.avg_pool (#4984)
authorpyjhzwh <pyjhzwh@gmail.com>
Sat, 7 Mar 2020 00:39:33 +0000 (19:39 -0500)
committerGitHub <noreply@github.com>
Sat, 7 Mar 2020 00:39:33 +0000 (09:39 +0900)
commitde0869de10bd353fc0daf6872a68677c6154483e
tree27a2bdbb17d0b5975aff62f87e2a4ca8de63dd44tree | snapshot
parente5044cb9c2d247506bc5aa0e04ce65ea31077179commit | diff
Fix stride default value None in torch.nn.functional.avg_pool (#4984)

* fix unordered dictionary problem for python version 3.5

* modify style

* default value of stride in torch.nn.functional.avg_pool is None

* delete prev modifications

* add testcase for nn.functional.avg_pool2d
python/tvm/relay/frontend/pytorch.py diff | blob | history
tests/python/frontend/pytorch/test_forward.py diff | blob | history
Domain: Machine Learning / ML Framework;