Nano: refactor inference optimizer code for better code reuse (#6330)
* add base optimizer for future InferenceOptimizer keras * fix type * add type change * change doc
This commit is contained in:
parent
c9b06eb431
commit
654baef1a3
1 changed files with 2 additions and 1 deletions
|
|
@ -15,7 +15,8 @@ bigdl.nano.pytorch.InferenceOptimizer
|
||||||
.. autoclass:: bigdl.nano.pytorch.InferenceOptimizer
|
.. autoclass:: bigdl.nano.pytorch.InferenceOptimizer
|
||||||
:members:
|
:members:
|
||||||
:undoc-members:
|
:undoc-members:
|
||||||
:exclude-members:
|
:exclude-members: ALL_INFERENCE_ACCELERATION_METHOD
|
||||||
|
:inherited-members:
|
||||||
|
|
||||||
bigdl.nano.pytorch.TorchNano
|
bigdl.nano.pytorch.TorchNano
|
||||||
---------------------------
|
---------------------------
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue