projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 917eac9) | patch
Keep weights name unchanged during SsaRewrite (#16932)
authorKimish Patel <kimishpatel@fb.com>
Mon, 11 Feb 2019 22:32:30 +0000 (14:32 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Mon, 11 Feb 2019 22:55:31 +0000 (14:55 -0800)
commit4292d13240e23a4a343b4ccb153214ab11c8d255
treed7a32939aec67de42ae4d2699fe0ce38f17e182ftree | snapshot
parent917eac91f4d8040eb51480fdfe1f75e99c33ac1acommit | diff
Keep weights name unchanged during SsaRewrite (#16932)

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

During onnxifi transformation net ssa is rewritten. At the last step the weight
names are changed back to what they were before. The diff keeps the weight
names unchanged thru the process.

Reviewed By: yinghai

Differential Revision: D13972597

fbshipit-source-id: 7c29857f788a674edf625c073b345f2b44267b33
caffe2/onnx/onnx_exporter.cc diff | blob | history
caffe2/onnx/onnx_exporter.h diff | blob | history
caffe2/opt/onnxifi_transformer.cc diff | blob | history
caffe2/opt/onnxifi_transformer.h diff | blob | history
caffe2/python/pybind_state.cc diff | blob | history
Domain: Machine Learning / ML Framework;