Skip to content
Snippets Groups Projects
Commit 1a166e59 authored by auphelia's avatar auphelia
Browse files

[Test] Commit quantized avg pool brevitas export after pre-commit

parent 82f87146
No related branches found
No related tags found
No related merge requests found
......@@ -64,9 +64,9 @@ def test_brevitas_avg_pool_export(kernel_size, stride, signed, bit_width):
)
b_avgpool.eval()
expected = b_avgpool.forward(input_quant_tensor).tensor.detach().numpy()
# finn execution
idict = {model.graph.input[0].name : inp}
idict = {model.graph.input[0].name: inp}
odict = oxe.execute_onnx(model, idict, True)
produced = odict[model.graph.output[0].name]
......
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