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

[Test] fix test streamline fc for new FC nets

parent e4dacb83
No related branches found
No related tags found
No related merge requests found
......@@ -82,7 +82,7 @@ def test_streamline_fc(size, wbits, abits):
model = model.transform(RemoveUnusedTensors())
assert len(model.graph.initializer) == 11
assert len(model.graph.value_info) == 21
assert len(model.graph.quantization_annotation) == 18
assert len(model.graph.quantization_annotation) == 20
produced_ctx = oxe.execute_onnx(model, input_dict, True)
produced = produced_ctx[model.graph.output[0].name]
assert np.isclose(expected, produced, atol=1e-3).all()
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