fix merge issue with constantfill

aaf0f9e1 · Khalique · 924aa8e5 · aaf0f9e1
Commit aaf0f9e1 authored Jan 24, 2019 by Khalique
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

src/onnx/onnx.cpp src/onnx/onnx.cpp +5 -0

No files found.
--- a/src/onnx/onnx.cpp
+++ b/src/onnx/onnx.cpp
@@ -599,6 +599,11 @@ struct onnx_parser
            input_as_shape = parse_value(attributes.at("input_as_shape")).at<int>();
        }

+        if(contains(attributes, "value"))
+        {
+            value = parse_value(attributes.at("value")).at<float>();
+        }
+
        if(contains(attributes, "extra_shape"))
        {
            MIGRAPHX_THROW("ConstantFill: cannot handle extra shape attribute");