2 Commits

Author	SHA1	Message	Date
Yinghai Lu	1d05d0d848	Improve onnxifi backend init time (#17375) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/17375 Previously we create the onnxGraph first and take it to the onnx manager for registration. It doesn't work well in practice. This diff takes "bring your own constructor" approach to reduce the resource spent doing backend compilation. Reviewed By: kimishpatel, rdzhabarov Differential Revision: D14173793 fbshipit-source-id: cbc4fe99fc522f017466b2fce88ffc67ae6757cf	2019-02-22 16:58:30 -08:00
Kimish Patel	e58cc6ab28	Enable single graph sharing between multiple threads for onnxifiop (#16047) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/16047 Implements single thead safe map enabling sharing of generated graph between different ops. Added model_id to every onnxified op to help create a unique id in the map. Some formatting fix. Reviewed By: yinghai Differential Revision: D13663927 fbshipit-source-id: 27417e8fe752fdd48abb6a87966cd76d592e1206	2019-01-16 12:19:16 -08:00