Fix epsilon bug

811c333d · turneram · 05db88b7 · 811c333d
Commit 811c333d authored Jun 01, 2022 by turneram
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

src/onnx/parse_layernorm.cpp src/onnx/parse_layernorm.cpp +2 -2

No files found.
--- a/src/onnx/parse_layernorm.cpp
+++ b/src/onnx/parse_layernorm.cpp
@@ -29,10 +29,10 @@ struct parse_layernorm : op_parser<parse_layernorm>
        if(contains(info.attributes, "epsilon"))
        {
            epsilon = parser.parse_value(info.attributes.at("epsilon")).at<float>();
-        }
+        } 
        if(contains(info.attributes, "axis"))
        {
-            epsilon = parser.parse_value(info.attributes.at("axis")).at<int64_t>();
+            axis = parser.parse_value(info.attributes.at("axis")).at<int64_t>();
        }

        auto epsilon_lit = info.add_literal(literal{shape{x_type, {1}}, {epsilon}});