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

[Test] add input quantizer parallelism to end2end tests

parent 942f4f19
No related branches found
No related tags found
No related merge requests found
......@@ -119,6 +119,10 @@ def fold_tfc(model):
fcl_inst.set_nodeattr("inFIFODepth", ififo)
fcl_inst.set_nodeattr("outFIFODepth", ofifo)
fcl_inst.set_nodeattr("ram_style", ramstyle)
# set parallelism for input quantizer to be same as first layer's SIMD
inp_qnt_node = model.get_nodes_by_op_type("Thresholding_Batch")[0]
inp_qnt = getCustomOp(inp_qnt_node)
inp_qnt.set_nodeattr("PE", 49)
return model
......
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