github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-04-17 03:00:54 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix: adhere to ONNX spec of kernel_shape

Browse Source 
ONNX spec:
"kernel_shape : list of ints
    The shape of the convolution kernel. If not present, should be
	inferred from input W."

We were taking the number of input/output feature maps into account,
which we realized we should not.
This commit is contained in:
youben11
2022-10-05 10:59:46 +01:00
committed by Ayoub Benaissa
parent dce0a86aa1
commit 7901d8868c
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/execution/test_convolution.py
View File

@@ -269,7 +269,7 @@ def test_conv2d(input_shape, weight_shape, group, strides, dilations, has_bias,
1,
"NOTSET",
ValueError,
"expected kernel_shape to be (1, 1, 2, 2), but got (1, 2)",
"expected kernel_shape to be (2, 2), but got (1, 2)",
),
pytest.param(
(1, 1, 4, 4),