Skip to content
Snippets Groups Projects
Commit 34fb09d4 authored by auphelia's avatar auphelia
Browse files

[Test] Fix bug in setting of idict

parent 07dd7e72
No related branches found
No related tags found
No related merge requests found
......@@ -65,7 +65,7 @@ def test_absorb_transp_into_flatten(perm, shape, ishape, data_layout):
inp_values = np.random.uniform(low=-1, high=1, size=tuple(ishape)).astype(
np.float32
)
idict = {"inp": inp_values}
idict = {model.graph.input[0].name: inp_values}
assert oxe.compare_execution(model, model_transformed, idict)
# only some of the parameter combinations lead to a graph that will be changed when
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment