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
This commit is contained in:
Junwei Deng 2022-10-19 14:29:24 +08:00 committed by GitHub
parent a0d6f860b8
commit 6a7ab76a8e
2 changed files with 17 additions and 1 deletions

View file

@ -24,3 +24,11 @@ bigdl.nano.pytorch.TorchNano
:members: :members:
:undoc-members: :undoc-members:
:exclude-members: run :exclude-members: run
Patch API
---------------------------
.. automodule:: bigdl.nano.pytorch.dispatcher
:members:
:undoc-members:
:show-inheritance:

View file

@ -21,4 +21,12 @@ bigdl.nano.tf.optimizers
--------------------------- ---------------------------
.. autoclass:: bigdl.nano.tf.optimizers.SparseAdam .. autoclass:: bigdl.nano.tf.optimizers.SparseAdam
:members: :members:
:undoc-members: :undoc-members:
Patch API
---------------------------
.. automodule:: bigdl.nano.tf.dispatcher
:members:
:undoc-members:
:show-inheritance: