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]` to apply when converting the model. E.g. `[Optimize.DEFAULT]`
representative_dataset: A representative dataset that can be used to representative_dataset: A representative dataset that can be used to
generate input and output samples for the model. The converter can use the 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. experimental_new_converter: Experimental flag, subject to change.
Enables MLIR-based conversion instead of TOCO conversion. Enables MLIR-based conversion instead of TOCO conversion.
experimental_new_quantizer: Experimental flag, subject to change. 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. rate for the exponentially weighted infinity norm.
epsilon: A small constant for numerical stability. epsilon: A small constant for numerical stability.
name: Optional name for the operations created when applying gradients. 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`, **kwargs: keyword arguments. Allowed to be {`clipnorm`, `clipvalue`, `lr`,
`decay`}. `clipnorm` is clip gradients by norm; `clipvalue` is clip `decay`}. `clipnorm` is clip gradients by norm; `clipvalue` is clip
gradients by value, `decay` is included for backward compatibility to gradients by value, `decay` is included for backward compatibility to