Serialization of experimental and custom ops (#2862)
authorJozef Daniecki <jozef.daniecki@intel.com>
Fri, 30 Oct 2020 04:31:27 +0000 (05:31 +0100)
committerGitHub <noreply@github.com>
Fri, 30 Oct 2020 04:31:27 +0000 (07:31 +0300)
commit34af04e1a80a74dd80747dd9a8e6112e859763bf
tree92ba5e37d5f0b16905533be193390bbe9a19a5f6
parent251429511da0d4777899d3830f102497eda35197
Serialization of experimental and custom ops (#2862)

* Add IEGeneric node type handling.

* Remove dependency on plugin_api library.

IEGeneric type name is passed via Visiotr API as new syntetic atribute.

* Add custom operations support

* Fix charachter literals comparison.

* Pass custom opsets to CNNNetwork:serialize().

IE extensions are stored in ngraph CNNNetwork and later used to pass
custom opsets to serialization transformation.

* Refactor custom ops tests to use template_extension library.

* Add comment on __generic_ie_type__ purpose.
16 files changed:
inference-engine/include/cpp/ie_cnn_network.h
inference-engine/src/inference_engine/cnn_network_ngraph_impl.cpp
inference-engine/src/inference_engine/cnn_network_ngraph_impl.hpp
inference-engine/src/inference_engine/generic_ie.cpp
inference-engine/src/plugin_api/generic_ie.hpp
inference-engine/src/readers/ir_reader/ie_ir_parser.cpp
inference-engine/src/readers/ir_reader/ie_ir_parser.hpp
inference-engine/src/readers/onnx_reader/ie_onnx_reader.cpp
inference-engine/src/transformations/include/transformations/serialize.hpp
inference-engine/src/transformations/src/transformations/serialize.cpp
inference-engine/tests/functional/inference_engine/ir_serialization/custom_ops.cpp [new file with mode: 0644]
inference-engine/tests/functional/inference_engine/ir_serialization/models/custom_op.prototxt [new file with mode: 0644]
inference-engine/tests/functional/inference_engine/ir_serialization/models/custom_op.xml [new file with mode: 0644]
inference-engine/tests/functional/inference_engine/ir_serialization/models/experimental_detectron_detection_output.xml [new file with mode: 0644]
inference-engine/tests/functional/inference_engine/ir_serialization/models/experimental_detectron_roi_feature_extractor.xml [new file with mode: 0644]
inference-engine/tests/functional/inference_engine/ir_serialization/serialize.cpp