Skip to content

Commit b5d1ec1

Browse files
committed
clarify ML framework for Extension API
1 parent c5164de commit b5d1ec1

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

docs/advanced/extension.rst

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,9 @@ To implement a custom layer in ``hls4ml`` with the extension API, the required c
1818
* Function config template
1919
* Registration of layer, source code, and templates
2020

21+
.. note::
22+
currently, then extension API supports keras models. Support for pytorch models is in development.
23+
2124
Complete example
2225
================
2326

0 commit comments

Comments
 (0)