From 5cb4b01a7f50ade32378fcb54a56da640369d9ba Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Wed, 30 Oct 2019 17:56:01 +0000
Subject: [PATCH] [Test] Added not runable test for model with mixed nodes
---
tests/test_mixed_onnx_exec.py | 22 ++++++++++++++++++++++
1 file changed, 22 insertions(+)
create mode 100644 tests/test_mixed_onnx_exec.py
diff --git a/tests/test_mixed_onnx_exec.py b/tests/test_mixed_onnx_exec.py
new file mode 100644
index 000000000..06fa7514f
--- /dev/null
+++ b/tests/test_mixed_onnx_exec.py
@@ -0,0 +1,22 @@
+import numpy as np
+from onnx import TensorProto, helper
+
+
+def test_execute_mixed_model():
+
+ out0 = helper.make_tensor_value_info("out", TensorProto.FLOAT, [6, 3, 2, 2])
+
+ graph_def = helper.make_graph(
+ nodes=[
+ helper.make_node(
+ "MultiThreshold", ["v", "thresholds"], ["out"], domain="finn"
+ ),
+ helper.make_node("Relu", ["out0"], ["out1"]),
+ ],
+ name="test-model",
+ inputs=[
+ helper.make_tensor_value_info("v", TensorProto.FLOAT, [6, 3, 2, 2]),
+ helper.make_tensor_value_info("thresholds", TensorProto.FLOAT, [3, 7]),
+ ],
+ outputs=[helper.make_tensor_value_info("out", TensorProto.FLOAT, [6, 3, 2, 2])],
+ )
--
GitLab