Skip to content
Snippets Groups Projects
Commit aa331d12 authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Test] catch streamlining fail in mnv1 end2end test

parent 5b832859
No related branches found
No related tags found
No related merge requests found
......@@ -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():
......
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