Merge pull request #37192 from ashutosh1919:tflite_piq

PiperOrigin-RevId: 301442417
Change-Id: Iec4bba1d74354ec0d873ff8477550021602e7aec
This commit is contained in:
TensorFlower Gardener 2020-03-17 13:39:23 -07:00
commit 83c0affbbd
2 changed files with 4 additions and 2 deletions

View File

@ -365,7 +365,9 @@ class TFLiteConverterV2(TFLiteConverterBase):
to apply when converting the model. E.g. `[Optimize.DEFAULT]`
representative_dataset: A representative dataset that can be used to
generate input and output samples for the model. The converter can use the
dataset to evaluate different optimizations.
dataset to evaluate different optimizations. Note that this is an optional
attribute but it is necessary if INT8 is the only support builtin ops in
target ops.
experimental_new_converter: Experimental flag, subject to change.
Enables MLIR-based conversion instead of TOCO conversion.
experimental_new_quantizer: Experimental flag, subject to change.

View File

@ -80,7 +80,7 @@ class Nadam(optimizer_v2.OptimizerV2):
rate for the exponentially weighted infinity norm.
epsilon: A small constant for numerical stability.
name: Optional name for the operations created when applying gradients.
Defaults to "Adamax".
Defaults to "Nadam".
**kwargs: keyword arguments. Allowed to be {`clipnorm`, `clipvalue`, `lr`,
`decay`}. `clipnorm` is clip gradients by norm; `clipvalue` is clip
gradients by value, `decay` is included for backward compatibility to