Skip to content

N
ngraph
Commit b56828fb authored by Jaikrishnan Menon's avatar Jaikrishnan Menon
Browse files
Options

CPU Direct Execution: Implement Floor, Negative and Sqrt

parent 0692dd8c
Hide whitespace changes
Inline Side-by-side
Showing with 18 additions and 0 deletions
src/ngraph/runtime/cpu/cpu_builder.cpp
View file @ b56828fb
...@@ -209,12 +209,30 @@ namespace ngraph ...@@ -209,12 +209,30 @@ namespace ngraph
BUILD_UNARY_ELEMWISE_FUNCTOR(runtime::cpu::kernel::ceil); 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 <> template <>
void Builder::BUILDER_DECL(ngraph::op::Relu) void Builder::BUILDER_DECL(ngraph::op::Relu)
{ {
BUILD_UNARY_ELEMWISE_FUNCTOR(runtime::cpu::kernel::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 <> template <>
void Builder::BUILDER_DECL(ngraph::op::Result) void Builder::BUILDER_DECL(ngraph::op::Result)
{ {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment