From 34fb09d4d2dd3e9db63e2d70ff9e30b59f62c982 Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Thu, 25 Jun 2020 15:47:31 +0100
Subject: [PATCH] [Test] Fix bug in setting of idict
---
tests/transformation/test_absorb_transp_into_flatten.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tests/transformation/test_absorb_transp_into_flatten.py b/tests/transformation/test_absorb_transp_into_flatten.py
index 609f28139..fbfa15277 100644
--- a/tests/transformation/test_absorb_transp_into_flatten.py
+++ b/tests/transformation/test_absorb_transp_into_flatten.py
@@ -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
--
GitLab