Fix for Conv op (#1556)

75a18827 · Michał Karzyński · GitHub · ae6a2903 · 75a18827
Unverified Commit 75a18827 authored Sep 04, 2018 by Michał Karzyński Committed by GitHub Sep 04, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

conv.cpp src/ngraph/frontend/onnx_import/op/conv.cpp +1 -1

No files found.
--- a/src/ngraph/frontend/onnx_import/op/conv.cpp
+++ b/src/ngraph/frontend/onnx_import/op/conv.cpp
@@ -106,7 +106,7 @@ namespace ngraph
                auto data = inputs.at(0);
                auto filters = inputs.at(1);

-                int groups{node.get_attribute_value<int>("group", 1)};
+                int64_t groups{node.get_attribute_value<int64_t>("group", 1)};

                // TODO: update to ASSERTION CHECK
                if (groups < 0 || groups > data->get_shape().at(1) ||