Fix copy/paste error on batchnorm op assignment

d6000754 · Jayaram Bobba · 9af9031e · d6000754
Commit d6000754 authored Mar 08, 2018 by Jayaram Bobba
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

cpu_assignment.cpp src/ngraph/runtime/cpu/pass/cpu_assignment.cpp +1 -1

No files found.
--- a/src/ngraph/runtime/cpu/pass/cpu_assignment.cpp
+++ b/src/ngraph/runtime/cpu/pass/cpu_assignment.cpp
@@ -230,7 +230,7 @@ namespace ngraph
                template <>
                void CPUAssignment::ASSIGN_DECL(ngraph::op::BatchNorm)
                {
-                    auto batchnorm = static_cast<op::ReluBackprop*>(node);
+                    auto batchnorm = static_cast<op::BatchNorm*>(node);
                    auto op_annotations =
                        std::make_shared<ngraph::runtime::cpu::CPUOpAnnotations>();
                    op_annotations->set_mkldnn_op(true);