projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 4ad17c9) | patch
JIT IR - Add option to remove prefix string when converting from JIT IR to NetDef...
authorDuc Ngo <duc@fb.com>
Tue, 12 Mar 2019 23:54:47 +0000 (16:54 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 13 Mar 2019 00:02:26 +0000 (17:02 -0700)
commit552f903c63355906ce46ca6cdba91d70589db501
tree7772a77a5d1d6aca7addfcbcf4906e616a92b667tree | snapshot
parent4ad17c9031680dc4d5997bc4999ea4a93d33c8d0commit | diff
JIT IR - Add option to remove prefix string when converting from JIT IR to NetDef (#17931)

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

When converting from NetDef to IR and back, the prefix string should be removed so the operator types are preserved in caffe2.

Reviewed By: ZolotukhinM

Differential Revision: D14425954

fbshipit-source-id: 2807e7337b0f804f126970768b1250a4a8c5f35c
test/cpp/jit/test_netdef_converter.h diff | blob | history
torch/csrc/jit/netdef_converter.cpp diff | blob | history
torch/csrc/jit/netdef_converter.h diff | blob | history
Domain: Machine Learning / ML Framework;