projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 3df79f4) | patch
Adding ONNX export for torch.expand and torch.ne (#15050)
authorSpandan Tiwari <sptiwari@microsoft.com>
Mon, 17 Dec 2018 21:45:21 +0000 (13:45 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Mon, 17 Dec 2018 21:48:14 +0000 (13:48 -0800)
commit700271d0e92c33a4cce3fee34b39ad5baa179b00
treebcd85e8c58ba3a0739bb716034c5a73ada98a5b7tree | snapshot
parent3df79f403e8b9621d5adb0447266becd10d633b0commit | diff
Adding ONNX export for torch.expand and torch.ne (#15050)

Summary:
`torch.expand` and `torch.ne` are used often in models and this PR adds ONNX export support for them. ArmenAg has created issue https://github.com/pytorch/pytorch/issues/10882 for this.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15050

Differential Revision: D13453036

Pulled By: houseroad

fbshipit-source-id: 4724b4ffcebda6cd6b2acac51d6733cb27318daf
test/onnx/expect/TestOperators.test_expand.expect [new file with mode: 0644] blob
test/onnx/expect/TestOperators.test_ne.expect [new file with mode: 0644] blob
test/onnx/test_operators.py diff | blob | history
test/test_jit.py diff | blob | history
torch/csrc/jit/export.cpp diff | blob | history
torch/onnx/symbolic.py diff | blob | history
Domain: Machine Learning / ML Framework;