From ea2477d56c002f144997a40284e0be96bf1a162a Mon Sep 17 00:00:00 2001 From: Taehee Jeong Date: Thu, 4 Feb 2021 22:35:17 -0800 Subject: [PATCH] Fix quantization debugger's `tf_export` This should be usable with TF v2, too. PiperOrigin-RevId: 355786139 Change-Id: I0725913ab3b410c23588cc59b5847f351a0eeb5a --- .../lite/experimental/quantization_debugger/debugger.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/lite/experimental/quantization_debugger/debugger.py b/tensorflow/lite/experimental/quantization_debugger/debugger.py index d4c744735ba..bfccd35a898 100644 --- a/tensorflow/lite/experimental/quantization_debugger/debugger.py +++ b/tensorflow/lite/experimental/quantization_debugger/debugger.py @@ -44,7 +44,7 @@ def _get_quant_params( return None -@tf_export.tf_export(v1=['lite.experimental.QuantizationDebugOptions']) +@tf_export.tf_export('lite.experimental.QuantizationDebugOptions') class QuantizationDebugOptions: """Debug options to set up a given QuantizationDebugger.""" @@ -71,7 +71,7 @@ class QuantizationDebugOptions: self.model_debug_metrics = model_debug_metrics -@tf_export.tf_export(v1=['lite.experimental.QuantizationDebugger']) +@tf_export.tf_export('lite.experimental.QuantizationDebugger') class QuantizationDebugger: """Debugger for Quantized TensorFlow Lite debug mode models.