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

Updated nGraph custom op documentation (#4605)

Browse Source 
* Updated nGraph custom op documentation

* Fixed comments
This commit is contained in:
Ilya Churaev
2021-03-10 16:17:08 +03:00
committed by GitHub
parent 95e3a1bf5d
commit efea25d28a
3 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/template_extension/op.cpp
View File

@@ -5,9 +5,9 @@
using namespace TemplateExtension;
constexpr ngraph::NodeTypeInfo Operation::type_info;
//! [op:ctor]
NGRAPH_RTTI_DEFINITION(TemplateExtension::Operation, "Template", 0);
Operation::Operation(const ngraph::Output<ngraph::Node> &arg, int64_t add) : Op({arg}), add(add) {
constructor_validate_and_infer_types();
}

3
docs/template_extension/op.hpp
View File

@@ -11,8 +11,7 @@ namespace TemplateExtension {
class Operation : public ngraph::op::Op {
public:
static constexpr ngraph::NodeTypeInfo type_info{"Template", 0};
const ngraph::NodeTypeInfo& get_type_info() const override { return type_info; }
NGRAPH_RTTI_DECLARATION;
Operation() = default;
Operation(const ngraph::Output<ngraph::Node>& arg, int64_t add);