fix onnx model

xadupre · Nov 16, 2023 · aa890b3 · aa890b3
1 parent c38451b
commit aa890b3
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/onnxruntime/test/testdata/custom_op_local_function/custom_gemm.h b/onnxruntime/test/testdata/custom_op_local_function/custom_gemm.h
@@ -1,12 +1,13 @@
 // Copyright (c) Microsoft Corporation. All rights reserved.
 // Licensed under the MIT License.
+#pragma once
 
 #define ORT_API_MANUAL_INIT
 #include <onnxruntime_c_api.h>
 #include <onnxruntime_cxx_api.h>
 #undef ORT_API_MANUAL_INIT
 
-#pragma once
+#include <vector>
 
 namespace Cpu {
 

diff --git a/onnxruntime/test/testdata/custom_op_local_function/custom_op_local_function.cc b/onnxruntime/test/testdata/custom_op_local_function/custom_op_local_function.cc
@@ -6,7 +6,6 @@
 #include <vector>
 #include <cmath>
 #include <mutex>
-#include <system_error>
 
 #include "core/common/common.h"
 #include "core/framework/ortdevice.h"

diff --git a/onnxruntime/test/testdata/custom_op_local_function/custom_ops_type_inference_fails_0.onnx b/onnxruntime/test/testdata/custom_op_local_function/custom_ops_type_inference_fails_0.onnx