Skip to content

Commit 863c57b

Browse files
committed
Fix extension test for Keras v3
1 parent a357b7a commit 863c57b

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

docs/advanced/extension.rst

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,10 @@ For concreteness, let's say our custom layer ``KReverse`` is implemented in Kera
3535
def call(self, inputs):
3636
return tf.reverse(inputs, axis=[-1])
3737
38+
def get_config(self):
39+
return super().get_config()
40+
41+
Make sure you define a ``get_config()`` method for your custom layer as this is needed for correct parsing.
3842
We can define the equivalent layer in hls4ml ``HReverse``, which inherits from ``hls4ml.model.layers.Layer``.
3943

4044
.. code-block:: Python

test/pytest/test_extensions.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,10 @@ def __init__(self):
1919
def call(self, inputs):
2020
return tf.reverse(inputs, axis=[-1])
2121

22+
def get_config(self):
23+
# Breaks serialization and parsing in hls4ml if not defined
24+
return super().get_config()
25+
2226

2327
# hls4ml layer implementation
2428
class HReverse(hls4ml.model.layers.Layer):

0 commit comments

Comments
 (0)