fastmachinelearning / hls4ml

Machine learning on FPGAs using HLS
https://fastmachinelearning.org/hls4ml
Apache License 2.0
1.24k stars 402 forks source link

Fix loading weights in n-dim dense -> 1x1 conv #821

Closed vloncar closed 1 year ago

vloncar commented 1 year ago

Description

In a change from using the readder to storing weights as attributes we missed the case of creating PointwiseConv nodes from n-dimensional Dense layer. This fixes it and adds a test for the future.

Type of change

Tests

Added a test in test_multi_dense.py

Checklist