projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 503f412) | patch
Enable single graph sharing between multiple threads for onnxifiop (#16047)
authorKimish Patel <kimishpatel@fb.com>
Wed, 16 Jan 2019 19:46:04 +0000 (11:46 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 16 Jan 2019 20:19:16 +0000 (12:19 -0800)
commite58cc6ab281e8e36b5780cfc285e8c219a532e0d
tree9c3e585fe4e7aee1877a0a613eed18acc0359e15tree | snapshot
parent503f412f795d6e91d3b16aafc1d522fbdc41b440commit | diff
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
caffe2/onnx/onnxifi_graph_info.cc [new file with mode: 0644] blob
caffe2/onnx/onnxifi_graph_info.h [new file with mode: 0644] blob
caffe2/onnx/onnxifi_init.cc diff | blob | history
caffe2/onnx/onnxifi_init.h diff | blob | history
caffe2/operators/onnxifi_op.h diff | blob | history
caffe2/opt/onnxifi_transformer.cc diff | blob | history
Domain: Machine Learning / ML Framework;