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:
|
||||
:undoc-members:
|
||||
: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
|
||||
:members:
|
||||
:undoc-members:
|
||||
:undoc-members:
|
||||
|
||||
Patch API
|
||||
---------------------------
|
||||
|
||||
.. automodule:: bigdl.nano.tf.dispatcher
|
||||
:members:
|
||||
:undoc-members:
|
||||
:show-inheritance:
|
||||
|
|
|
|||
Loading…
Reference in a new issue