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: