experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix quantization debugger's tf_export

Browse Source 
This should be usable with TF v2, too.

PiperOrigin-RevId: 355786139
Change-Id: I0725913ab3b410c23588cc59b5847f351a0eeb5a
This commit is contained in:
Taehee Jeong 2021-02-04 22:35:17 -08:00 committed by TensorFlower Gardener
parent 7b7dc64203
commit ea2477d56c
1 changed files with 2 additions and 2 deletions

4
tensorflow/lite/experimental/quantization_debugger/debugger.py
View File

@ -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.