diff --git a/tests/end2end/test_end2end_mobilenet_v1.py b/tests/end2end/test_end2end_mobilenet_v1.py index 79263a7099b91fb0dbaa10871f7859690ab9e4c2..5bfe8e1ea1b48ed77c40a584d624cc0ecdedb668 100644 --- a/tests/end2end/test_end2end_mobilenet_v1.py +++ b/tests/end2end/test_end2end_mobilenet_v1.py @@ -188,6 +188,10 @@ def test_end2end_mobilenet_streamline(): model = model.transform(GiveReadableTensorNames()) model = model.transform(InferDataTypes()) model.save(build_dir + "/end2end_mobilenet_streamlined.onnx") + assert ( + len(model.get_nodes_by_op_type("Add")) == 1 + ) # only final quantized bias Add op remains + assert len(model.get_nodes_by_op_type("Mul")) == 0 # no Mul ops remain def test_end2end_mobilenet_lowering():