diff --git a/Tools/ML/model.cxx b/Tools/ML/model.cxx index 7646c3ab2dc..bf15ab21221 100644 --- a/Tools/ML/model.cxx +++ b/Tools/ML/model.cxx @@ -222,7 +222,8 @@ void OnnxModel::modelSurgery(std::string inpath, std::string outpath, std::unord } } -void OnnxModel::add_concat_to_input(onnx::ModelProto& model, std::unordered_map originals) { +void OnnxModel::add_concat_to_input(onnx::ModelProto& model, std::unordered_map originals) +{ auto* graph = model.mutable_graph(); // 1. Save original input name and remove it diff --git a/Tools/ML/model.h b/Tools/ML/model.h index 1cace876019..4b58a927eaa 100644 --- a/Tools/ML/model.h +++ b/Tools/ML/model.h @@ -22,8 +22,8 @@ #include -#include #include +#include #include #include