diff --git a/docs/readthedocs/source/doc/PythonAPI/Nano/hpo_api.rst b/docs/readthedocs/source/doc/PythonAPI/Nano/hpo_api.rst
index 88a5b71d..e19d43ba 100644
--- a/docs/readthedocs/source/doc/PythonAPI/Nano/hpo_api.rst
+++ b/docs/readthedocs/source/doc/PythonAPI/Nano/hpo_api.rst
@@ -19,14 +19,14 @@ HPO for Tensorflow
 bigdl.nano.automl.tf.keras.Model
 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
 
-.. autoclass:: bigdl.nano.automl.tf.keras.Model.Model
+.. autoclass:: bigdl.nano.automl.tf.keras.Model
     :members: search, search_summary
 
 
 bigdl.nano.automl.tf.keras.Sequential
 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
 
-.. autoclass:: bigdl.nano.automl.tf.keras.Sequential.Sequential
+.. autoclass:: bigdl.nano.automl.tf.keras.Sequential
     :members: search, search_summary
 
 
diff --git a/docs/readthedocs/source/doc/PythonAPI/Nano/pytorch.rst b/docs/readthedocs/source/doc/PythonAPI/Nano/pytorch.rst
index a9d6d29a..c267da73 100644
--- a/docs/readthedocs/source/doc/PythonAPI/Nano/pytorch.rst
+++ b/docs/readthedocs/source/doc/PythonAPI/Nano/pytorch.rst
@@ -10,7 +10,7 @@ bigdl.nano.pytorch.Trainer
     :exclude-members: accelerator_connector, checkpoint_connector, reload_dataloaders_every_n_epochs, limit_val_batches, logger, logger_connector, state
 
 bigdl.nano.pytorch.InferenceOptimizer
----------------------------
+---------------------------------------
 
 .. autoclass:: bigdl.nano.pytorch.InferenceOptimizer
     :members:
@@ -21,23 +21,24 @@ bigdl.nano.pytorch.InferenceOptimizer
 TorchNano API
 ---------------------------
 
-.. automodule:: bigdl.nano.pytorch.torch_nano
+.. autoclass:: bigdl.nano.pytorch.TorchNano
     :members:
     :undoc-members:
     :exclude-members: run
 
+.. autofunction:: bigdl.nano.pytorch.nano
+
 Patch API
 ---------------------------
 
-.. automodule:: bigdl.nano.pytorch.dispatcher
-    :members:
-    :undoc-members:
-    :show-inheritance:
+.. autofunction:: bigdl.nano.pytorch.patch_torch
 
-.. automodule:: bigdl.nano.pytorch.patching.gpu_cpu.gpu_cpu
-    :members: patch_cuda, unpatch_cuda
-    :show-inheritance:
+.. autofunction:: bigdl.nano.pytorch.unpatch_torch
 
-.. automodule:: bigdl.nano.pytorch.patching.dtype_patching.dtype_patching
-    :members: patch_dtype
-    :show-inheritance:
+.. autofunction:: bigdl.nano.pytorch.patching.patch_cuda
+
+.. autofunction:: bigdl.nano.pytorch.patching.unpatch_cuda
+
+.. autofunction:: bigdl.nano.pytorch.patching.patch_dtype
+    
+.. autofunction:: bigdl.nano.pytorch.patching.patch_encryption
\ No newline at end of file
diff --git a/docs/readthedocs/source/doc/PythonAPI/Nano/tensorflow.rst b/docs/readthedocs/source/doc/PythonAPI/Nano/tensorflow.rst
index 3c95c854..8052747a 100644
--- a/docs/readthedocs/source/doc/PythonAPI/Nano/tensorflow.rst
+++ b/docs/readthedocs/source/doc/PythonAPI/Nano/tensorflow.rst
@@ -13,11 +13,6 @@ bigdl.nano.tf.keras
     :undoc-members:
     :inherited-members: Sequential
 
-.. autoclass:: bigdl.nano.tf.keras.InferenceOptimizer
-    :members:
-    :undoc-members:
-    :exclude-members: ALL_INFERENCE_ACCELERATION_METHOD
-
 .. autoclass:: bigdl.nano.tf.keras.layers.Embedding
     :members:
     :undoc-members:
@@ -28,10 +23,18 @@ bigdl.nano.tf.optimizers
     :members: 
     :undoc-members:
 
+bigdl.nano.tf.keras.InferenceOptimizer
+---------------------------------------
+.. autoclass:: bigdl.nano.tf.keras.InferenceOptimizer
+    :members:
+    :undoc-members:
+    :exclude-members: ALL_INFERENCE_ACCELERATION_METHOD
+
 Patch API
 ---------------------------
 
-.. automodule:: bigdl.nano.tf.dispatcher
-    :members:
-    :undoc-members:
-    :show-inheritance:
+.. autofunction:: bigdl.nano.tf.patch_tensorflow
+
+.. autofunction:: bigdl.nano.tf.unpatch_tensorflow
+
+.. autofunction:: bigdl.nano.tf.keras.nano_bf16
\ No newline at end of file