06cb1971c7
* Small fixes to (un)patch_tensorflow api doc and make its import path in api doc show as the recommended one * Add api doc for nano_bf16 decorator * Move api doc for bigdl.nano.tf.keras.InferenceOptimizer out of bigdl.nano.tf.keras to make it more clear * Fix python styles * Fix path in Nano PyTorch API docs * Add api doc for bigdl.nano.pytorch.patching.patch_encryption * Add a note box for bigdl.nano.pytorch.patching.patch_encryption api doc * Fix Python style again * Fix path in Nano HPO API doc and other small fixes
40 lines
No EOL
941 B
ReStructuredText
40 lines
No EOL
941 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.layers.Embedding
|
|
:members:
|
|
:undoc-members:
|
|
|
|
bigdl.nano.tf.optimizers
|
|
---------------------------
|
|
.. autoclass:: bigdl.nano.tf.optimizers.SparseAdam
|
|
:members:
|
|
:undoc-members:
|
|
|
|
bigdl.nano.tf.keras.InferenceOptimizer
|
|
---------------------------------------
|
|
.. autoclass:: bigdl.nano.tf.keras.InferenceOptimizer
|
|
:members:
|
|
:undoc-members:
|
|
:exclude-members: ALL_INFERENCE_ACCELERATION_METHOD
|
|
|
|
Patch API
|
|
---------------------------
|
|
|
|
.. autofunction:: bigdl.nano.tf.patch_tensorflow
|
|
|
|
.. autofunction:: bigdl.nano.tf.unpatch_tensorflow
|
|
|
|
.. autofunction:: bigdl.nano.tf.keras.nano_bf16 |