Add support for Fused ops to transformation passes (#3784)

7b618f03 · Michał Karzyński · Scott Cyphers · ac646533 · 7b618f03 · 7b618f03
Commit 7b618f03 authored 5 years ago by Michał Karzyński Committed by Scott Cyphers 5 years ago
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

opset0_downgrade.cpp src/ngraph/pass/opset0_downgrade.cpp +2 -0

opset1_upgrade.cpp src/ngraph/pass/opset1_upgrade.cpp +2 -0

No files found.
--- a/src/ngraph/pass/opset0_downgrade.cpp
+++ b/src/ngraph/pass/opset0_downgrade.cpp
@@ -33,12 +33,14 @@ using namespace ngraph;
 #define NGRAPH_OP(a, b) a,
 enum class OP_TYPEID
 {
+#include "ngraph/op/fused_op_tbl.hpp"
 #include "ngraph/op/op_tbl.hpp"
 };
 #undef NGRAPH_OP

 #define NGRAPH_OP(a, b) {#a, OP_TYPEID::a},
 static unordered_map<string, OP_TYPEID> typeid_map{
+#include "ngraph/op/fused_op_tbl.hpp"
 #include "ngraph/op/op_tbl.hpp"
 };
 #undef NGRAPH_OP

--- a/src/ngraph/pass/opset1_upgrade.cpp
+++ b/src/ngraph/pass/opset1_upgrade.cpp
@@ -42,12 +42,14 @@ using namespace ngraph;
 #define NGRAPH_OP(a, b) a,
 enum class OP_TYPEID
 {
+#include "ngraph/op/fused_op_tbl.hpp"
 #include "ngraph/op/op_tbl.hpp"
 };
 #undef NGRAPH_OP

 #define NGRAPH_OP(a, b) {#a, OP_TYPEID::a},
 static unordered_map<string, OP_TYPEID> typeid_map{
+#include "ngraph/op/fused_op_tbl.hpp"
 #include "ngraph/op/op_tbl.hpp"
 };
 #undef NGRAPH_OP