Xilinx / finn-base

Open Source Compiler Framework using ONNX as Frontend and IR
https://finn-base.readthedocs.io/
BSD 3-Clause "New" or "Revised" License
29 stars 17 forks source link

Changes for supporting non-equal dilation #29

Closed mmrahorovic closed 3 years ago

mmrahorovic commented 3 years ago

[im2col, lower_convs_to_matmul, change_3d_tensors_to_4d]: added support for non-equal dilation value along (H, W) dimensions. Also changed the tensor sparsity annotation from integer to list of integers. This is needed for non-square image support for the InferVVAU transformation.

[test_im2col, test_conv_lowering]: added test cases for non-equal dilation configurations.

[test_modelwrapper, test_general_transformation, test_merge_onnx_models]: changed tensor sparsity annotation for consistency (list of integers instead of integer).

maltanar commented 3 years ago

Thanks Mirza!