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
|
||||
:members:
|
||||
:undoc-members:
|
||||
:exclude-members:
|
||||
:exclude-members: ALL_INFERENCE_ACCELERATION_METHOD
|
||||
:inherited-members:
|
||||
|
||||
bigdl.nano.pytorch.TorchNano
|
||||
---------------------------
|
||||
|
|
|
|||
Loading…
Reference in a new issue