projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 97eb139) | patch
ONNX Export Argmin and Argmax ops
authorLara Haidar-Ahmad <lahaidar@microsoft.com>
Wed, 6 Mar 2019 20:05:48 +0000 (12:05 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 6 Mar 2019 20:11:47 +0000 (12:11 -0800)
commit073634612f0a1c88ae5155cfdadaf5e351f60149
treebe8556ffac22e870118ecbb9199fe8442a81080atree | snapshot
parent97eb139a942b34ac6c46159c33aee4f22292625ecommit | diff
ONNX Export Argmin and Argmax ops

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/17382

Differential Revision: D14338811

Pulled By: houseroad

fbshipit-source-id: be07548d8063d1aa94f1801c18137738365b85fb
test/onnx/expect/TestOperators.test_argmax.expect [new file with mode: 0644] blob
test/onnx/test_operators.py diff | blob | history
test/onnx/test_pytorch_onnx_caffe2.py diff | blob | history
torch/onnx/symbolic.py diff | blob | history
Domain: Machine Learning / ML Framework;