error ORT_INVALID_GRAPH(10) in C++

[LaurentBerger]

Hi,

python 3.10
onnx 1.15.0
onnxruntime 1.17.1
onnxruntime-tools 1.7.0
onnxscript 0.1.0.dev20230625 C:\lib\onnxscript
onnxsim 0.4.33

I created a model with one node in python :

import numpy as np
import onnx
import onnx.reference

shape_ini = (1, 2, 3, 4, 5)
data = np.arange(0, shape_ini[0]* shape_ini[1]*shape_ini[2]*shape_ini[3]*shape_ini[4]).reshape(shape_ini).astype(dtype=np.float32)    
shape_ini = data.shape 
select_axes = np.array([2], dtype=np.int64)
keepdims = 0
print("input DATA ")
print(data)
onnx_name = "testReduceMean2"
out1 = onnx.helper.make_tensor_value_info('out1', onnx.TensorProto.FLOAT, [1,2,4,5])
inp0 = onnx.helper.make_tensor_value_info('inp0', onnx.TensorProto.FLOAT, [1, 2, 3, 4, 5])
node1 = onnx.helper.make_node( "ReduceMean", inputs=["inp0"], outputs=["out1"], keepdims=keepdims , axes=select_axes)
graph = onnx.helper.make_graph([node1], onnx_name,  [inp0], [out1])
onnx_model = onnx.helper.make_model(graph)

feeds = {'inp0': data}
sess = onnx.reference.ReferenceEvaluator(onnx_model)
res_onnx = sess.run(["out1"], feeds)
print("ONNX result")
print(res_onnx[0])
print("Writting model")
with open(onnx_name + ".onnx", "wb") as f:
    f.write(onnx_model.SerializeToString())

Now I tried to read this model in C++ using ONNX sample with onnx-runtime 1.18.0:

#ifdef _WIN32
int wmain(int argc, ORTCHAR_T* argv[]) {
#else
int main(int argc, ORTCHAR_T* argv[]) {
#endif
  if (argc != 2) {
    std::cout << "Usage: ./onnx-api-example <onnx_model.onnx>" << std::endl;
    return -1;
  }

  std::basic_string<ORTCHAR_T> model_file = argv[1];
  // onnxruntime setup
  Ort::Env env(ORT_LOGGING_LEVEL_WARNING, "example-model-explorer");
  Ort::SessionOptions session_options;
  Ort::Session session = Ort::Session(env, model_file.c_str(), session_options);

I have got an error
error_code = ORT_INVALID_GRAPH (10)

stack trace is

onnxsample.exe!Ort::detail::ThrowStatus(const Ort::Status & st) Line 27 C++
onnxsample.exe!Ort::ThrowOnError(OrtStatus * ort_status) Line 34 C++
onnxsample.exe!Ort::Session::Session(const Ort::Env & env, const wchar_t * model_path, const Ort::SessionOptions & options) Line 1074 C++
onnxsample.exe!wmain(int argc, wchar_t * * argv) Line 70 C++

With my C++ code I can't read onnx model but model created with my python code seems wrong only in C++.
What's wrong in my python or C++ code?

[LaurentBerger]

I think it is relative to #14959