CPU Direct Execution: Implement Floor, Negative and Sqrt

b56828fb · Jaikrishnan Menon · 0692dd8c · b56828fb
Commit b56828fb authored Jul 20, 2018 by Jaikrishnan Menon
Hide whitespace changes
Inline Side-by-side

Showing with 18 additions and 0 deletions

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

No files found.
--- a/src/ngraph/runtime/cpu/cpu_builder.cpp
+++ b/src/ngraph/runtime/cpu/cpu_builder.cpp
@@ -209,12 +209,30 @@ namespace ngraph
                BUILD_UNARY_ELEMWISE_FUNCTOR(runtime::cpu::kernel::ceil);
            }

+            template <>
+            void Builder::BUILDER_DECL(ngraph::op::Floor)
+            {
+                BUILD_UNARY_ELEMWISE_FUNCTOR(runtime::cpu::kernel::floor);
+            }
+
+            template <>
+            void Builder::BUILDER_DECL(ngraph::op::Negative)
+            {
+                BUILD_UNARY_ELEMWISE_FUNCTOR(runtime::cpu::kernel::negative);
+            }
+
            template <>
            void Builder::BUILDER_DECL(ngraph::op::Relu)
            {
                BUILD_UNARY_ELEMWISE_FUNCTOR(runtime::cpu::kernel::relu);
            }

+            template <>
+            void Builder::BUILDER_DECL(ngraph::op::Sqrt)
+            {
+                BUILD_UNARY_ELEMWISE_FUNCTOR(runtime::cpu::kernel::sqrt);
+            }
+
            template <>
            void Builder::BUILDER_DECL(ngraph::op::Result)
            {