Skip to content

N
ngraph
Commit 86b783c6 authored by Pruthvi's avatar Pruthvi Committed by Scott Cyphers
Browse files
Options

fix failing bn test (#2175) 

* fix fialing bn test

* fix style
parent 05c7fbe4
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
src/ngraph/runtime/cpu/cpu_emitter.cpp
View file @ 86b783c6
...@@ -698,8 +698,8 @@ namespace ngraph ...@@ -698,8 +698,8 @@ namespace ngraph
if (args.size() == 3) if (args.size() == 3)
{ {
writer << "reference::batch_norm_three_outputs(" writer << "reference::batch_norm_training(" << batchnorm->get_eps_value()
<< batchnorm->get_eps_value() << ",\n"; << ",\n";
writer << " " << args[0].get_name() << ",\n"; writer << " " << args[0].get_name() << ",\n";
writer << " " << args[1].get_name() << ",\n"; writer << " " << args[1].get_name() << ",\n";
writer << " " << args[2].get_name() << ",\n"; writer << " " << args[2].get_name() << ",\n";
...@@ -710,7 +710,7 @@ namespace ngraph ...@@ -710,7 +710,7 @@ namespace ngraph
} }
else else
{ {
writer << "reference::batch_norm_one_output(" << batchnorm->get_eps_value() writer << "reference::batch_norm_inference(" << batchnorm->get_eps_value()
<< ",\n"; << ",\n";
writer << " " << args[0].get_name() << ",\n"; writer << " " << args[0].get_name() << ",\n";
writer << " " << args[1].get_name() << ",\n"; writer << " " << args[1].get_name() << ",\n";
...@@ -736,7 +736,7 @@ namespace ngraph ...@@ -736,7 +736,7 @@ namespace ngraph
const ngraph::op::BatchNormInference* batchnorm = const ngraph::op::BatchNormInference* batchnorm =
static_cast<const ngraph::op::BatchNormInference*>(node); static_cast<const ngraph::op::BatchNormInference*>(node);
writer << "reference::batch_norm_one_output(" << batchnorm->get_eps_value() writer << "reference::batch_norm_inference(" << batchnorm->get_eps_value()
<< ",\n"; << ",\n";
writer << " " << args[0].get_name() << ",\n"; writer << " " << args[0].get_name() << ",\n";
writer << " " << args[1].get_name() << ",\n"; writer << " " << args[1].get_name() << ",\n";
......
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