Replace count() with contains() for maps

Author: AndreySorokin7

app/Graph/build.cpp

@@ -227,8 +227,8 @@ void build_graph(it_lab_ai::Graph& graph, it_lab_ai::Tensor& input,
   try {
     std::sort(connection_list.begin(), connection_list.end(),
               [&](const auto& a, const auto& b) {
-                if (!name_to_layer_ptr.count(a.first) ||
-                    !name_to_layer_ptr.count(b.first)) {
+                if (!name_to_layer_ptr.contains(a.first) ||
+                    !name_to_layer_ptr.contains(b.first)) {
                   return false;
                 }
                 return name_to_layer_ptr[a.first]->getID() <
@@ -239,8 +239,8 @@ void build_graph(it_lab_ai::Graph& graph, it_lab_ai::Tensor& input,
   }
 
   for (const auto& [source_name, target_name] : connection_list) {
-    if (name_to_layer_ptr.count(source_name) &&
-        name_to_layer_ptr.count(target_name)) {
+    if (name_to_layer_ptr.contains(source_name) &&
+        name_to_layer_ptr.contains(target_name)) {
       if (target_name.find("Concat") != std::string::npos ||
           name_to_layer_ptr[target_name]->getName() == it_lab_ai::kConcat) {
         if (concat_connections.find(target_name) != concat_connections.end()) {
@@ -573,7 +573,7 @@ ParseResult parse_json_model(RuntimeOptions options,
             std::string constant_name = inputs[1].get<std::string>();
             constant_name = get_base_layer_name(constant_name);
 
-            if (layer_parameters.count(constant_name)) {
+            if (layer_parameters.contains(constant_name)) {
               splits = layer_parameters[constant_name];
             } else if (constant_name.find("onnx::") != std::string::npos) {
               splits = last_constant_value;
@@ -771,7 +771,7 @@ ParseResult parse_json_model(RuntimeOptions options,
             std::string constant_name = inputs[1].get<std::string>();
             constant_name = get_base_layer_name(constant_name);
 
-            if (layer_parameters.count(constant_name)) {
+            if (layer_parameters.contains(constant_name)) {
               shape = layer_parameters[constant_name];
             }
           }
@@ -833,7 +833,7 @@ ParseResult parse_json_model(RuntimeOptions options,
             std::string constant_name = inputs[1].get<std::string>();
             constant_name = get_base_layer_name(constant_name);
 
-            if (layer_parameters.count(constant_name)) {
+            if (layer_parameters.contains(constant_name)) {
               axes = layer_parameters[constant_name];
             } else if (constant_name.find("onnx::") != std::string::npos) {
               axes = last_constant_value;