Commit Graph

3 Commits

Author SHA1 Message Date
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
Marat Dukhan
45f0d05202 Adapt OnnxifiOp to removed suffix handling in ONNXIFI loader (#9571)
Summary:
Adapt to changes in onnx/onnx#1203
Pull Request resolved: https://github.com/pytorch/pytorch/pull/9571

Reviewed By: yinghai

Differential Revision: D8907892

Pulled By: bddppq

fbshipit-source-id: 9f88471639dbe9050194e84340f335bece834d5d
2018-07-18 19:26:23 -07:00
Yinghai Lu
c2a89b69b9 Support to ONNXIFI op (#8749)
Summary:
This PR adds basic support to ONNXIFI op.
Closes https://github.com/pytorch/pytorch/pull/8749

Reviewed By: Maratyszcza

Differential Revision: D8665739

Pulled By: yinghai

fbshipit-source-id: 961916f9e1a4a26390b73c4b648d177883143a22
2018-06-29 09:10:26 -07:00