ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Nano : support InferenceOptimizer for keras (#6458)

Browse source 
* add optimizer.py

* fix style

* set version for mypy

* add thread control for quantize, support openvino pot

* update optimizer

* fix typo

* update workflow

* temp commit

* modify ut

* refactor optimizer

* rebase, update based on comments
This commit is contained in:
Ruonan Wang 2022-11-18 09:39:12 +08:00 committed by GitHub
parent 022d8d661c
commit 11cc260980
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
docs/readthedocs/source/doc/PythonAPI/Nano/tensorflow.rst
View file

@ -13,6 +13,10 @@ bigdl.nano.tf.keras
:undoc-members:
:inherited-members: Sequential
.. autoclass:: bigdl.nano.tf.keras.InferenceOptimizer
:members:
:undoc-members:
.. autoclass:: bigdl.nano.tf.keras.layers.Embedding
:members:
:undoc-members: