fastmachinelearning / hls4ml

Machine learning on FPGAs using HLS

https://fastmachinelearning.org/hls4ml

Apache License 2.0

1.19k stars 390 forks source link

Closed vloncar closed 1 year ago

vloncar commented 1 year ago

Description

Support keepdims argument of pooling layers. Helps correctly parse some models.

Look in #715 for a test case.

jmduarte commented 1 year ago

Tests fail because our 0.2 testing image had TensorFlow 2.4.1 before keepdims was added to the config of these pooling layers (I think).

I created a 0.4 testing image with TensorFlow 2.9.1, so I relaunched tests with that.