projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 6084908) | patch
Fix interpolate trace (#18875)
authorElias Ellison <eellison@fb.com>
Sat, 6 Apr 2019 00:52:12 +0000 (17:52 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Sat, 6 Apr 2019 00:55:23 +0000 (17:55 -0700)
commite6bbbb017e1ad829316e7d7ea0f9ca9e78421da2
tree91568316be861cf06e36ad34a401db09c41c701dtree | snapshot
parent6084908287ab88b73f8af8d6469aac0c57cb1b43commit | diff
Fix interpolate trace (#18875)

Summary:
Fixes https://github.com/pytorch/pytorch/issues/10654

The issue is that in tracing `.size` returns an int tensor, and when an int tensor is multiplied by a scalar the int dominates and the scalar gets casted 0.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18875

Differential Revision: D14814441

Pulled By: eellison

fbshipit-source-id: a4e96a2698f2fcbf3ec4b2bb4c43a30250f30ad9
test/test_jit.py diff | blob | history
torch/nn/functional.py diff | blob | history
Domain: Machine Learning / ML Framework;