projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: ffc7158) | patch
Improved onnx export for 3 onnx ops. (#18512)
authorBenoit Steiner <benoitsteiner@fb.com>
Thu, 28 Mar 2019 15:52:01 +0000 (08:52 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 28 Mar 2019 15:55:21 +0000 (08:55 -0700)
commiteee760dbd341127cd9e33f04813532fbf9e63316
tree40bda869fdf314101804fa3c12ca4e9c7937c6f8tree | snapshot
parentffc7158bf2f97916305217e4203ef846c00161cecommit | diff
Improved onnx export for 3 onnx ops. (#18512)

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

Ceil and Floor have been supported since version 6 of ONNX: export them using the native onnx ops instead of an Aten op.
Similarly, support for the Where op has been added in version 9, so we don't need to wrap these op in an Aten op.

Reviewed By: houseroad

Differential Revision: D14635130

fbshipit-source-id: d54a2b6e295074a6214b5939b21051a6735c9958
caffe2/onnx/backend.cc diff | blob | history
caffe2/onnx/backend.h diff | blob | history
caffe2/python/onnx/tests/onnx_backend_test.py diff | blob | history
torch/onnx/symbolic.py diff | blob | history
Domain: Machine Learning / ML Framework;