projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 01be9b7) | patch
AD support for adaptive_avg_pool2d (#15459)
authorAiling Zhang <ailzhang@fb.com>
Fri, 21 Dec 2018 23:32:44 +0000 (15:32 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Fri, 21 Dec 2018 23:38:24 +0000 (15:38 -0800)
commit70aafad08a9e01f4de04cfe6a486556c1c201a58
treeee402491623fc8482850856de1102c0845619ef1tree | snapshot
parent01be9b7292d7763e2e8e2ae1016bd2539d5bb083commit | diff
AD support for adaptive_avg_pool2d (#15459)

Summary:
This adds AD support for adaptive_avg_pool2d, which is necessary for resnet50 in pytorch/vision:master. cc: soumith asuhan dlibenzi

apaszke  I saw that autodiff bug you fixed in #15403 , as it doesn't prevent this PR from passing, so I'll leave it for your PR to fix it. :)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15459

Differential Revision: D13534732

Pulled By: ailzhang

fbshipit-source-id: 4e48b93e35d5ecfe7bd64b6a132a55b07843f206
test/test_jit.py diff | blob | history
torch/csrc/jit/symbolic_script.cpp diff | blob | history
Domain: Machine Learning / ML Framework;