Skip to content
Snippets Groups Projects
Unverified Commit 9d7de204 authored by Yaman Umuroglu's avatar Yaman Umuroglu Committed by GitHub
Browse files

[Test] respect mem_mode in test_end2end_cnv_w1a1

parent 9ccb1ddb
No related branches found
No related tags found
No related merge requests found
......@@ -112,8 +112,8 @@ def test_end2end_cnv_w1a1_streamline():
def test_end2end_cnv_w1a1_convert_to_hls_layers():
model = ModelWrapper(build_dir + "/end2end_cnv_w1a1_streamlined.onnx")
model = model.transform(to_hls.InferBinaryStreamingFCLayer())
model = model.transform(to_hls.InferQuantizedStreamingFCLayer())
model = model.transform(to_hls.InferBinaryStreamingFCLayer(mem_mode))
model = model.transform(to_hls.InferQuantizedStreamingFCLayer(mem_mode))
model = model.transform(to_hls.InferConvInpGen())
model = model.transform(to_hls.InferStreamingMaxPool())
model = model.transform(MoveReshape())
......
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