修改获取参数的方法

zjhellofss · Apr 21, 2024 · 3bb5a7b · 3bb5a7b
1 parent fb715cf
commit 3bb5a7b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/source/layer/details/flatten.cpp b/source/layer/details/flatten.cpp
@@ -119,12 +119,12 @@ StatusCode FlattenLayer::CreateInstance(const std::shared_ptr<RuntimeOperator>&
     return StatusCode::kParseParameterError;
   }
 
-  if (params.find("end_dim") == params.end()) {
+  if (!op->has_parameter("end_dim")) {
     LOG(ERROR) << "Can not find the dimension parameter";
     return StatusCode::kParseParameterError;
   }
 
-  if (params.find("start_dim") == params.end()) {
+  if (!op->has_parameter("start_dim")) {
     LOG(ERROR) << "Can not find the dimension parameter";
     return StatusCode::kParseParameterError;
   }