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

[Test] fix test_convert_to_hls_layers

parent 122c89cd
No related branches found
No related tags found
No related merge requests found
......@@ -89,7 +89,6 @@ def test_convert_to_hls_layers_tfc_w1a1():
assert fc3.op_type == "StreamingFCLayer_Batch"
assert model.get_tensor_shape(fc3.input[0]) == [1, 64]
assert model.get_tensor_shape(fc3.input[1]) == [64, 10]
os.remove(export_onnx_path)
fc0w = getCustomOp(fc0)
fc0w.set_nodeattr("SIMD", 784)
......
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