saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-29 03:30:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix nested test

Browse source
This commit is contained in:
Jingyan Wang 2025-02-05 22:05:07 +00:00
parent bb393a987b
commit 8fc043f474
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
onnxruntime/test/python/onnxruntime_test_python_nested_control_flow_op.py
View file

@ -205,8 +205,8 @@ def make_opt_nested_greater_or_equal() -> ModelProto:
m = helper.make_model(
graph,
opset_imports=[
# helper.make_opsetid("", 17),
helper.make_opsetid("", 14),
helper.make_opsetid("", 17),
# helper.make_opsetid("", 15),
],
)
@ -245,10 +245,10 @@ def test_nested_optional_greater_or_equal_with_ep_context() -> None:
m = make_opt_nested_greater_or_equal()
# Create session options and add config entries
so = SessionOptions()
so.add_config_entry("ep.context_enable", "1")
so.add_config_entry("ep.context_file_path", "EP_Context_model.onnx")
so.add_config_entry("ep.context_embed_mode", "0")
so = ort.SessionOptions()
so.add_session_config_entry("ep.context_enable", "1")
so.add_session_config_entry("ep.context_file_path", "EP_Context_model.onnx")
so.add_session_config_entry("ep.context_embed_mode", "0")
providers = ["TensorrtExecutionProvider"]
session = ort.InferenceSession(