Added BWD macro for all new ops (#9356)

This commit is contained in:
Ilya Churaev authored and GitHub committed 2021-12-23 07:19:34 +03:00
1 parent 3d244a41ab
commit 16b39d15d0
10 files changed
+11

No files matched your search

@@ -70,6 +70,7 @@ public:
using Attributes = op::util::DetectionOutputBase::AttributesBase;
OPENVINO_OP("DetectionOutput", "opset8", op::util::DetectionOutputBase);
BWDCMP_RTTI_DECLARATION;
DetectionOutput() = default;
/// \brief Constructs a DetectionOutput operation
@@ -14,6 +14,7 @@ namespace v8 {
class OPENVINO_API MatrixNms : public util::NmsBase {
public:
OPENVINO_OP("MatrixNms", "opset8", op::util::NmsBase);
BWDCMP_RTTI_DECLARATION;
enum class DecayFunction { GAUSSIAN, LINEAR };
@@ -63,6 +63,7 @@ namespace v8 {
class OPENVINO_API MaxPool : public op::util::MaxPoolBase {
public:
OPENVINO_OP("MaxPool", "opset8", op::util::MaxPoolBase);
BWDCMP_RTTI_DECLARATION;
/// \brief Constructs an empty MaxPool operation.
MaxPool() = default;
@@ -14,6 +14,7 @@ namespace v8 {
class OPENVINO_API MulticlassNms : public util::NmsBase {
public:
OPENVINO_OP("MulticlassNms", "opset8", op::util::NmsBase);
BWDCMP_RTTI_DECLARATION;
/// \brief Structure that specifies attributes of the operation
struct Attributes {
+1
View File
@@ -51,6 +51,7 @@ namespace v8 {
class OPENVINO_API Softmax : public Op {
public:
OPENVINO_OP("Softmax", "opset8");
BWDCMP_RTTI_DECLARATION;
Softmax() = default;
/// \brief Constructs a softmax operation.
+1
View File
@@ -64,6 +64,7 @@ bool ov::op::v0::DetectionOutput::visit_attributes(AttributeVisitor& visitor) {
}
// ------------------------------ V8 ------------------------------
BWDCMP_RTTI_DEFINITION(ov::op::v8::DetectionOutput);
ov::op::v8::DetectionOutput::DetectionOutput(const Output<Node>& box_logits,
const Output<Node>& class_preds,
const Output<Node>& proposals,
+1
View File
@@ -17,6 +17,7 @@
#include "ngraph/util.hpp"
using namespace ngraph;
BWDCMP_RTTI_DEFINITION(ov::op::v8::MatrixNms);
op::v8::MatrixNms::MatrixNms() : NmsBase(m_attrs.output_type, m_attrs.nms_top_k, m_attrs.keep_top_k) {}
+1
View File
@@ -16,6 +16,7 @@ using namespace std;
using namespace ngraph;
BWDCMP_RTTI_DEFINITION(op::v1::MaxPool);
BWDCMP_RTTI_DEFINITION(op::v8::MaxPool);
op::v1::MaxPool::MaxPool(const Output<Node>& arg,
const Strides& strides,
+2
View File
@@ -18,6 +18,8 @@
using namespace ngraph;
BWDCMP_RTTI_DEFINITION(ov::op::v8::MulticlassNms);
op::v8::MulticlassNms::MulticlassNms() : NmsBase(m_attrs.output_type, m_attrs.nms_top_k, m_attrs.keep_top_k) {}
op::v8::MulticlassNms::MulticlassNms(const Output<Node>& boxes, const Output<Node>& scores, const Attributes& attrs)
+1
View File
@@ -101,6 +101,7 @@ bool op::v1::Softmax::has_evaluate() const {
}
// *** SOFTMAX OP SET V8 ***
BWDCMP_RTTI_DEFINITION(op::v8::Softmax);
op::v8::Softmax::Softmax(const Output<Node>& arg, const int64_t axis) : Op({arg}), m_axis(axis) {
constructor_validate_and_infer_types();