projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 75d6d88) | patch
add an utility function to check whether it's in the middle of onnx export or not
authorLu Fang <lufang@fb.com>
Tue, 9 Apr 2019 17:01:48 +0000 (10:01 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Tue, 9 Apr 2019 17:07:08 +0000 (10:07 -0700)
commitba77eadbca3b7d31d8125246f434e7ff7cca1597
treec779be3f7be2ed06d750c7703f46c6373cd10d11tree | snapshot
parent75d6d8833d4f5cd5eaa33b4fe68ac7db6527507fcommit | diff
add an utility function to check whether it's in the middle of onnx export or not

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

Reviewed By: yinghai

Differential Revision: D14849878

Pulled By: houseroad

fbshipit-source-id: a0a4a57f5f9f315ba1334edfccc9284a8099d17f
test/onnx/test_utility_funs.py [new file with mode: 0644] blob
torch/onnx/__init__.py diff | blob | history
torch/onnx/utils.py diff | blob | history
Domain: Machine Learning / ML Framework;