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:
parent
a0d6f860b8
commit
6a7ab76a8e
2 changed files with 17 additions and 1 deletions
|
|
@ -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:
|
||||||
|
|
|
||||||
|
|
@ -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:
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue