Merge pull request #46704 from AdamHillier:patch-1

PiperOrigin-RevId: 355653955 Change-Id: Ia604b27c13903430b4ce7ae85299aecfea6853bc
2021-02-04 10:58:44 -08:00 · 2021-02-04 10:58:44 -08:00 · 766a32278d
commit 766a32278d
parent b109f7c0ac 9db9d08855
1 changed files with 1 additions and 0 deletions
--- a/tensorflow/lite/python/lite.py
+++ b/tensorflow/lite/python/lite.py
@ -411,6 +411,7 @@ class QuantizationMode(object):
    """Checks if the graph contains any training-time quantization ops."""
    training_quant_ops = frozenset({
        "FakeQuantWithMinMaxVars", "FakeQuantWithMinMaxVarsPerChannel",
+        "FakeQuantWithMinMaxArgs", "FakeQuantWithMinMaxArgsPerChannel",
        "QuantizeAndDequantizeV2", "QuantizeAndDequantizeV3"
    })