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):
202
202
203
203
self ._assert_outputs_equal (quantized_model , same_architecture_model )
204
204
205
- def testSerialization_TF2SavedModel (self ):
205
+ def testSerialization_SavedModel (self ):
206
206
if compat .is_v1_apis ():
207
207
return
208
208
@@ -216,24 +216,6 @@ def testSerialization_TF2SavedModel(self):
216
216
217
217
self ._assert_outputs_equal (quantized_model , loaded_model )
218
218
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
-
237
219
def testSerialization_TFCheckpoint (self ):
238
220
model = test_utils .build_simple_dense_model ()
239
221
quantized_model = quantize .quantize_model (model )
You can’t perform that action at this time.
0 commit comments