Added BWD macro for all new ops (#9356)
This commit is contained in:
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 {
|
||||
|
||||
@@ -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.
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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) {}
|
||||
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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)
|
||||
|
||||
@@ -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();
|
||||
|
||||
Reference in new issue
Block a user