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
if (args.size() == 3)
{
writer << "reference::batch_norm_three_outputs("
<< batchnorm->get_eps_value() << ",\n";
writer << "reference::batch_norm_training(" << batchnorm->get_eps_value()
<< ",\n";
writer << " " << args[0].get_name() << ",\n";
writer << " " << args[1].get_name() << ",\n";
writer << " " << args[2].get_name() << ",\n";
......@@ -710,7 +710,7 @@ namespace ngraph
}
else
{
writer << "reference::batch_norm_one_output(" << batchnorm->get_eps_value()
writer << "reference::batch_norm_inference(" << batchnorm->get_eps_value()
<< ",\n";
writer << " " << args[0].get_name() << ",\n";
writer << " " << args[1].get_name() << ",\n";
......@@ -736,7 +736,7 @@ namespace ngraph
const ngraph::op::BatchNormInference* batchnorm =
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";
writer << " " << args[0].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