File tree Expand file tree Collapse file tree 1 file changed +1
-19
lines changed
tensorflow_model_optimization/python/core/quantization/keras Expand file tree Collapse file tree 1 file changed +1
-19
lines changed Original file line number Diff line number Diff line change @@ -202,7 +202,7 @@ def testSerialization_KerasCheckpoint(self):
202202
203203 self ._assert_outputs_equal (quantized_model , same_architecture_model )
204204
205- def testSerialization_TF2SavedModel (self ):
205+ def testSerialization_SavedModel (self ):
206206 if compat .is_v1_apis ():
207207 return
208208
@@ -216,24 +216,6 @@ def testSerialization_TF2SavedModel(self):
216216
217217 self ._assert_outputs_equal (quantized_model , loaded_model )
218218
219- def testSerialization_TF1SavedModel (self ):
220- if not compat .is_v1_apis ():
221- return
222-
223- model = test_utils .build_simple_dense_model ()
224- quantized_model = quantize .quantize_model (model )
225- self ._train_model (quantized_model )
226-
227- saved_model_dir = tempfile .mkdtemp ()
228- with quantize .quantize_scope ():
229- tf .keras .experimental .export_saved_model (quantized_model , saved_model_dir )
230-
231- with quantize .quantize_scope ():
232- loaded_model = tf .keras .experimental .load_from_saved_model (
233- saved_model_dir )
234-
235- self ._assert_outputs_equal (quantized_model , loaded_model )
236-
237219 def testSerialization_TFCheckpoint (self ):
238220 model = test_utils .build_simple_dense_model ()
239221 quantized_model = quantize .quantize_model (model )
You can’t perform that action at this time.
0 commit comments