diff --git a/src/ngraph/runtime/cpu/cpu_emitter.cpp b/src/ngraph/runtime/cpu/cpu_emitter.cpp index 1bd1b733c8fb87c812bed72ae62921cfd59aacd4..3cb3576b8c4fb134022d21fb99a6e217a6f13b61 100644 --- a/src/ngraph/runtime/cpu/cpu_emitter.cpp +++ b/src/ngraph/runtime/cpu/cpu_emitter.cpp @@ -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";