projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: e04c919) | patch
Fix ONNX symbolic for argmin and argmax (#18261)
authorLu Fang <lufang@fb.com>
Thu, 21 Mar 2019 05:45:57 +0000 (22:45 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 21 Mar 2019 05:51:13 +0000 (22:51 -0700)
commit2016daaf5150c6bd52101698c93843c8abc8b3ef
tree5a7aa30f65a5153b8f0997d5498f1608c4f95f54tree | snapshot
parente04c9195b746926a72be8af73b60fd996ec62403commit | diff
Fix ONNX symbolic for argmin and argmax (#18261)

Summary:
Fix the problem introduced in https://github.com/pytorch/pytorch/pull/17103
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18261

Reviewed By: bddppq

Differential Revision: D14558781

Pulled By: houseroad

fbshipit-source-id: 7bb50072e77d1d7b2a93f4011fa1362f26e9df1c
test/onnx/expect/TestOperators.test_argmax.expect diff | blob | history
torch/onnx/symbolic.py diff | blob | history
Domain: Machine Learning / ML Framework;