IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add docs for OPENVINO_FRAMEWORK_MAP macro (#14928)

Browse Source 
* Add docs for OPENVINO_FRAMEWORK_MAP macro

Ticket: 98762

* Apply suggestions from code review

Co-authored-by: Piotr Krzemiński <piotrkrzeminski1234@gmail.com>

---------

Co-authored-by: Piotr Krzemiński <piotrkrzeminski1234@gmail.com>
This commit is contained in:
Mateusz Tabaka
2023-03-26 20:00:43 +02:00
committed by GitHub
parent 6eb8f4b2b7
commit 1df14c6a6c
3 changed files with 65 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
docs/snippets/ov_extensions.cpp
View File

@@ -17,6 +17,12 @@
#include <openvino/opsets/opset8.hpp>
//! [frontend_extension_ThresholdedReLU_header]
//! [frontend_extension_framework_map_macro_headers]
#include <openvino/frontend/extension/op.hpp>
#include <openvino/frontend/onnx/extension/op.hpp>
#include <openvino/frontend/tensorflow/extension/op.hpp>
//! [frontend_extension_framework_map_macro_headers]
#include <identity.hpp>
//! [frontend_extension_CustomOperation]
@@ -40,6 +46,27 @@ public:
std::shared_ptr<ov::Node> clone_with_new_inputs(const ov::OutputVector&) const override { return nullptr; }
};
//! [frontend_extension_framework_map_macro_CustomOp]
class CustomOp : public ov::op::Op {
std::string m_mode;
int m_axis;
public:
OPENVINO_OP("CustomOp");
OPENVINO_FRAMEWORK_MAP(onnx, "CustomOp", { {"m_mode", "mode"} }, { {"m_axis", -1} });
OPENVINO_FRAMEWORK_MAP(tensorflow, "CustomOpV3", { {"m_axis", "axis"} }, { {"m_mode", "linear"} });
bool visit_attributes(ov::AttributeVisitor& visitor) override {
visitor.on_attribute("m_mode", m_mode);
visitor.on_attribute("m_axis", m_axis);
return true;
}
// ... implement other required methods
//! [frontend_extension_framework_map_macro_CustomOp]
std::shared_ptr<ov::Node> clone_with_new_inputs(const ov::OutputVector&) const override { return nullptr; }
};
int main() {
{
//! [add_extension]
@@ -125,5 +152,12 @@ ov::Core core;
core.add_extension("openvino_template_extension.so");
//! [add_extension_lib]
}
{
//! [frontend_extension_framework_map_macro_add_extension]
ov::Core core;
core.add_extension(ov::frontend::OpExtension<CustomOp>());
//! [frontend_extension_framework_map_macro_add_extension]
}
return 0;
}