projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 586d030) | patch
Remove StopGradient op when it is inplace in inference (#12152)
authorwuhuikx <hui.h.wu@intel.com>
Sat, 12 Jan 2019 07:52:28 +0000 (23:52 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Sat, 12 Jan 2019 07:55:01 +0000 (23:55 -0800)
commit35480a7c444470e753226b9a2d4e766650f90590
tree18cb001fd58e1bbbb4a512a7680266b3ecb6e691tree | snapshot
parent586d0303113123f59c1992c57714bec60b90fd76commit | diff
Remove StopGradient op when it is inplace in inference (#12152)

Summary:
For Inference, if the StopGradient op is inpalce, we just remove it.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/12152

Differential Revision: D13633946

Pulled By: yinghai

fbshipit-source-id: 57762bcc37b38a1d39cb4af316ca50bfe961b105
caffe2/opt/optimize_ideep.cc diff | blob | history
Domain: Machine Learning / ML Framework;