Fix topk deserialize to be backwards compatible (#4172)

NervanaSystems · Jan 14, 2020 · ac27139 · ac27139
1 parent bc4a08f
commit ac27139
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/ngraph/serializer.cpp b/src/ngraph/serializer.cpp
@@ -2945,7 +2945,8 @@ shared_ptr<Node> JSONDeserializer::deserialize_node(json node_js)
         {
             auto compute_max = node_js.at("compute_max").get<bool>();
             auto target_type = read_element_type(node_js.at("index_element_type"));
-            op::TopKSortType sort = node_js.at("sort").get<op::TopKSortType>();
+            op::TopKSortType sort =
+                get_or_default<op::TopKSortType>(node_js, "sort", op::TopKSortType::SORT_VALUES);
             if (has_key(node_js, "top_k_axis"))
             {
                 auto top_k_axis = node_js.at("top_k_axis").get<size_t>();