projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 1442808) | patch
add fbgemm fp16 (fbfcpacked) support, add global_init_net in predictor_export_meta...
authorWeiyi Zheng <wyz@fb.com>
Wed, 20 Mar 2019 20:45:07 +0000 (13:45 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 20 Mar 2019 20:52:10 +0000 (13:52 -0700)
commit1b71f6d4ebfc7b736a54111578d9e9edb5607c5f
tree9ab27e6e7e66d9322b246e120c86b2cf0cefa5fdtree | snapshot
parent1442808fcdb3fba7eae1122ad4fc6f4e2cb8df15commit | diff
add fbgemm fp16 (fbfcpacked) support, add global_init_net in predictor_export_meta (#17905)

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

support adding op in global_init_net. because pred_init_net is per thread, and just doesn't cut it.

Reviewed By: jspark1105

Differential Revision: D14114134

fbshipit-source-id: 112bb2ceb9d3d5e663dd430585567f4eaa2db35f
caffe2/python/predictor/predictor_exporter.py diff | blob | history
caffe2/python/predictor/predictor_exporter_test.py diff | blob | history
Domain: Machine Learning / ML Framework;