Skip to content

N
ngraph
Commit 8686eb7a authored by Matthew Brookhart's avatar Matthew Brookhart Committed by Scott Cyphers
Browse files
Options

block off softmax to avoid compilation errors from redefining variables (#802)

parent 1d80cabe
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/ngraph/runtime/cpu/cpu_emitter.cpp
View file @ 8686eb7a
...@@ -3331,6 +3331,7 @@ namespace ngraph ...@@ -3331,6 +3331,7 @@ namespace ngraph
template <> template <>
void CPU_Emitter::EMITTER_DECL(ngraph::op::Softmax) void CPU_Emitter::EMITTER_DECL(ngraph::op::Softmax)
{ {
writer.block_begin();
const ngraph::op::Softmax* softmax = static_cast<const ngraph::op::Softmax*>(node); const ngraph::op::Softmax* softmax = static_cast<const ngraph::op::Softmax*>(node);
auto type = out[0].get_type(); auto type = out[0].get_type();
auto shape = out[0].get_shape(); auto shape = out[0].get_shape();
...@@ -3509,6 +3510,7 @@ namespace ngraph ...@@ -3509,6 +3510,7 @@ namespace ngraph
writer.block_end(); writer.block_end();
} }
} }
writer.block_end();
} }
template <> template <>
......
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