saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/caffe2/opt
History
Yinghai Lu 98b333d810 Tag model_id and onnxifi index in OnnxifiOp (#16648) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/16648

We added onnxGraph sharing keyed on model id and net seq number but we forgot to supply these info to the Onnxifi. Therefore, we will only create ONE onnxGraph whatsoever... This diff adds necessary info to the OnnxifiOp to prevent this from happening.

Reviewed By: bertmaher, rdzhabarov

Differential Revision: D13912356

fbshipit-source-id: fe8982327287a35f32fe3b125d94b617d18c0ab5
2019-02-01 18:51:51 -08:00
..
annotations.cc Add support for device annotations on blobs 2018-09-28 14:11:54 -07:00
annotations.h Add support for device annotations on blobs 2018-09-28 14:11:54 -07:00
backend_cutting.cc Windows shared build (#13550) 2018-11-16 12:16:28 -08:00
backend_cutting.h Replace direct include of caffe2.pb.h with an intermediary header caffe2_pb.h (#10946) 2018-08-28 11:57:08 -07:00
backend_cutting_test.cc Windows shared build (#13550) 2018-11-16 12:16:28 -08:00
bound_shape_inference_test.cc Bound shape inference for c2 (#16081) 2019-01-16 19:02:56 -08:00
bound_shape_inferencer.cc Bound shape inference for c2 (#16081) 2019-01-16 19:02:56 -08:00
bound_shape_inferencer.h Bound shape inference for c2 (#16081) 2019-01-16 19:02:56 -08:00
CMakeLists.txt
converter.cc Relax check on outputs (#15458) 2018-12-21 14:19:37 -08:00
converter.h Fix namespace typo. (#16021) 2019-01-15 16:43:08 -08:00
converter_nomigraph_test.cc nomnigraph - easy - use new test utils in converter_nomnigraph_test (#15751) 2019-01-14 18:38:38 -08:00
dead_code_elim.cc Add dead code elimination pass (#10101) 2018-09-05 23:55:17 -07:00
dead_code_elim_test.cc Add dead code elimination pass (#10101) 2018-09-05 23:55:17 -07:00
device.cc fixes to device.cc (#11122) 2018-08-31 09:25:26 -07:00
device.h More fixes for hidden visibility (#10624) 2018-08-20 10:11:59 -07:00
device_test.cc Windows shared build (#13550) 2018-11-16 12:16:28 -08:00
distributed.cc Open source distributed code (#12254) 2018-10-03 21:41:14 -07:00
distributed.h Open source distributed code (#12254) 2018-10-03 21:41:14 -07:00
distributed_converter.cc Open source distributed code (#12254) 2018-10-03 21:41:14 -07:00
distributed_test.cc Open source distributed code (#12254) 2018-10-03 21:41:14 -07:00
fusion.cc Enable junk fill for the default CPU allocator (#13377) 2018-11-08 00:02:37 -08:00
fusion.h Unify all *_EXPORT and *_IMPORT macros across c++ backend (#12019) 2018-09-25 17:41:05 -07:00
mobile.cc
mobile.h More fixes for hidden visibility (#10624) 2018-08-20 10:11:59 -07:00
mobile_test.cc Windows shared build (#13550) 2018-11-16 12:16:28 -08:00
onnx_convert.h Roll nomnigraph build into caffe2 (#11303) 2018-09-06 19:38:09 -07:00
onnxifi_transformer.cc Tag model_id and onnxifi index in OnnxifiOp (#16648) 2019-02-01 18:51:51 -08:00
onnxifi_transformer.h Tag model_id and onnxifi index in OnnxifiOp (#16648) 2019-02-01 18:51:51 -08:00
optimize_ideep.cc Remove StopGradient op when it is inplace in inference (#12152) 2019-01-11 23:55:01 -08:00
optimize_ideep.h Replace direct include of caffe2.pb.h with an intermediary header caffe2_pb.h (#10946) 2018-08-28 11:57:08 -07:00
optimizer.cc
optimizer.h Replace direct include of caffe2.pb.h with an intermediary header caffe2_pb.h (#10946) 2018-08-28 11:57:08 -07:00
passes.cc Move registry fully to c10 (#12077) 2018-09-27 03:09:54 -07:00
passes.h Move registry fully to c10 (#12077) 2018-09-27 03:09:54 -07:00
sink.cc Unify all *_EXPORT and *_IMPORT macros across c++ backend (#12019) 2018-09-25 17:41:05 -07:00
sink.h Replace direct include of caffe2.pb.h with an intermediary header caffe2_pb.h (#10946) 2018-08-28 11:57:08 -07:00