Skip to content

Commit 3c2c2e3

Browse files
k-w-wtensorflower-gardener
authored andcommitted
Remove internal calls to keras.experimental.export_saved_model.
PiperOrigin-RevId: 341149480
1 parent ae4d198 commit 3c2c2e3

File tree

1 file changed

+1
-19
lines changed

1 file changed

+1
-19
lines changed

tensorflow_model_optimization/python/core/quantization/keras/quantize_integration_test.py

Lines changed: 1 addition & 19 deletions
Original file line numberDiff line numberDiff 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)

0 commit comments

Comments
 (0)