11cc260980
* add optimizer.py * fix style * set version for mypy * add thread control for quantize, support openvino pot * update optimizer * fix typo * update workflow * temp commit * modify ut * refactor optimizer * rebase, update based on comments
36 lines
755 B
ReStructuredText
36 lines
755 B
ReStructuredText
Nano Tensorflow API
|
|
==================
|
|
|
|
bigdl.nano.tf.keras
|
|
---------------------------
|
|
|
|
.. autoclass:: bigdl.nano.tf.keras.Model
|
|
:members: fit, quantize, trace
|
|
:undoc-members:
|
|
|
|
.. autoclass:: bigdl.nano.tf.keras.Sequential
|
|
:members:
|
|
:undoc-members:
|
|
:inherited-members: Sequential
|
|
|
|
.. autoclass:: bigdl.nano.tf.keras.InferenceOptimizer
|
|
:members:
|
|
:undoc-members:
|
|
|
|
.. autoclass:: bigdl.nano.tf.keras.layers.Embedding
|
|
:members:
|
|
:undoc-members:
|
|
|
|
bigdl.nano.tf.optimizers
|
|
---------------------------
|
|
.. autoclass:: bigdl.nano.tf.optimizers.SparseAdam
|
|
:members:
|
|
:undoc-members:
|
|
|
|
Patch API
|
|
---------------------------
|
|
|
|
.. automodule:: bigdl.nano.tf.dispatcher
|
|
:members:
|
|
:undoc-members:
|
|
:show-inheritance:
|