From 11cc2609805f8350cb0100ffe71e5c53b9ae6f24 Mon Sep 17 00:00:00 2001 From: Ruonan Wang <105281011+rnwang04@users.noreply.github.com> Date: Fri, 18 Nov 2022 09:39:12 +0800 Subject: [PATCH] Nano : support InferenceOptimizer for keras (#6458) * 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 --- docs/readthedocs/source/doc/PythonAPI/Nano/tensorflow.rst | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/docs/readthedocs/source/doc/PythonAPI/Nano/tensorflow.rst b/docs/readthedocs/source/doc/PythonAPI/Nano/tensorflow.rst index a0bbb19c..1a3e10eb 100644 --- a/docs/readthedocs/source/doc/PythonAPI/Nano/tensorflow.rst +++ b/docs/readthedocs/source/doc/PythonAPI/Nano/tensorflow.rst @@ -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: