[ DTS ] Reduces (#940)

2020-06-18 11:36:07 +03:00
parent 88cccee0b7
commit bb44f17a06
6 changed files with 377 additions and 2 deletions
--- a/ngraph/src/ngraph/validation_util.cpp
+++ b/ngraph/src/ngraph/validation_util.cpp
@@ -849,8 +849,11 @@ int64_t ngraph::normalize_axis(const std::string& node_description,
    }

    const auto tensor_rank_value = tensor_rank.get_length();
-    return normalize_axis(
-        node_description, axis, tensor_rank_value, -tensor_rank_value, tensor_rank_value - 1);
+    return normalize_axis(node_description,
+                          axis,
+                          tensor_rank_value,
+                          -tensor_rank_value,
+                          tensor_rank_value ? (tensor_rank_value - 1) : 0);
 }

 int64_t ngraph::normalize_axis(const Node* node,