CPU Direct Execution: Implement Subtract

e2d1b5bc · Jaikrishnan Menon · 6576932f · e2d1b5bc
Commit e2d1b5bc authored Jul 20, 2018 by Jaikrishnan Menon
Show whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

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

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

+            template <>
+            void Builder::BUILDER_DECL(ngraph::op::Subtract)
+            {
+                BUILD_BINARY_ELEMWISE_FUNCTOR(runtime::cpu::kernel::subtract);
+            }
+
            template <>
            void Builder::BUILDER_DECL(ngraph::op::Multiply)
            {