fixed equality for beta=1.0

7d72a081 · Khalique · 8c2b3895 · 7d72a081
Commit 7d72a081 authored Dec 11, 2018 by Khalique
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/onnx/onnx.cpp src/onnx/onnx.cpp +1 -1

No files found.
--- a/src/onnx/onnx.cpp
+++ b/src/onnx/onnx.cpp
@@ -428,7 +428,7 @@ struct onnx_parser
            {
                auto l3 = prog.add_instruction(op::dot{alpha}, l1, l2);
                auto l4 = args[2];
-                if(beta == 1.f)
+                if(beta != 1.f)
                {
                    auto beta_val = prog.add_literal(beta);
                    auto l5 = prog.add_instruction(op::scalar{args[2]->get_shape()}, beta_val);