Add constant folding for Power for CPU (#3849)

f543849e · Sayantan Sarkar · Scott Cyphers · 3b558069 · f543849e
Commit f543849e authored Nov 05, 2019 by Sayantan Sarkar Committed by Scott Cyphers Nov 05, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 0 deletions

cpu_builder.cpp src/ngraph/runtime/cpu/cpu_builder.cpp +7 -0

No files found.
--- a/src/ngraph/runtime/cpu/cpu_builder.cpp
+++ b/src/ngraph/runtime/cpu/cpu_builder.cpp
@@ -495,6 +495,12 @@ namespace ngraph
                BUILD_BINARY_ELEMWISE_CF_FUNCTOR(runtime::cpu::kernel::multiply);
            }

+            template <>
+            NodeExecutorTy Builder::BUILDER_CF_DECL(ngraph::op::Power)
+            {
+                BUILD_BINARY_ELEMWISE_CF_FUNCTOR(runtime::cpu::kernel::cwise_pow);
+            }
+
            template <>
            NodeExecutorTy Builder::BUILDER_CF_DECL(ngraph::op::Divide)
            {
@@ -727,6 +733,7 @@ namespace ngraph
                REGISTER_CF_BUILDER(Xor);
                REGISTER_CF_BUILDER(Sign);
                REGISTER_CF_BUILDER(Not);
+                REGISTER_CF_BUILDER(Power);
            }
        }
    }