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

[Test] also check StreamingFC gen for final layer for lfcw1a1

parent 83f7deeb
No related branches found
No related tags found
No related merge requests found
......@@ -50,4 +50,8 @@ def test_convert_to_hls_layers_lfc_w1a1():
assert model.get_tensor_shape(fc2.input[0]) == [1, 1024]
assert model.get_tensor_shape(fc2.input[1]) == [1024, 1024]
assert model.get_tensor_shape(fc2.input[2]) == [1024, 1]
fc3 = model.graph.node[5]
assert fc3.op_type == "StreamingFCLayer_Batch"
assert model.get_tensor_shape(fc3.input[0]) == [1, 1024]
assert model.get_tensor_shape(fc3.input[1]) == [1024, 10]
os.remove(export_onnx_path)
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