From 6a7ab76a8e3d04738f90d0dcb5bbd1dfc44fdf84 Mon Sep 17 00:00:00 2001 From: Junwei Deng <35031544+TheaperDeng@users.noreply.github.com> Date: Wed, 19 Oct 2022 14:29:24 +0800 Subject: [PATCH] Nano: add a patch_torch/patch_tensorflow API for friendly usage (#5946) * add a prototype code * add patch_nano * add pep8 changes * add cv ppl * fix bug * add unpatch * add updates * update comment * add changes * add doc * update and refactor * update test * fix doc * fix according to cmt * fix style * fix doc * fix bug * fix style * fix bug * fix code * add split patching api * fix doc * patch api to each framework * fix pep8 * fix style * fix style and cmt --- docs/readthedocs/source/doc/PythonAPI/Nano/pytorch.rst | 8 ++++++++ .../source/doc/PythonAPI/Nano/tensorflow.rst | 10 +++++++++- 2 files changed, 17 insertions(+), 1 deletion(-) diff --git a/docs/readthedocs/source/doc/PythonAPI/Nano/pytorch.rst b/docs/readthedocs/source/doc/PythonAPI/Nano/pytorch.rst index b7d65305..44de6b92 100644 --- a/docs/readthedocs/source/doc/PythonAPI/Nano/pytorch.rst +++ b/docs/readthedocs/source/doc/PythonAPI/Nano/pytorch.rst @@ -24,3 +24,11 @@ bigdl.nano.pytorch.TorchNano :members: :undoc-members: :exclude-members: run + +Patch API +--------------------------- + +.. automodule:: bigdl.nano.pytorch.dispatcher + :members: + :undoc-members: + :show-inheritance: diff --git a/docs/readthedocs/source/doc/PythonAPI/Nano/tensorflow.rst b/docs/readthedocs/source/doc/PythonAPI/Nano/tensorflow.rst index 059476df..a0bbb19c 100644 --- a/docs/readthedocs/source/doc/PythonAPI/Nano/tensorflow.rst +++ b/docs/readthedocs/source/doc/PythonAPI/Nano/tensorflow.rst @@ -21,4 +21,12 @@ bigdl.nano.tf.optimizers --------------------------- .. autoclass:: bigdl.nano.tf.optimizers.SparseAdam :members: - :undoc-members: \ No newline at end of file + :undoc-members: + +Patch API +--------------------------- + +.. automodule:: bigdl.nano.tf.dispatcher + :members: + :undoc-members: + :show-inheritance: