We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 70287e2 commit d8a5cc4Copy full SHA for d8a5cc4
tensorflow_model_analysis/model_util.py
@@ -590,8 +590,6 @@ def construct_fn(): # pylint: disable=invalid-name
590
model.register_add_metric_callbacks(add_metrics_callbacks)
591
model.graph_finalize()
592
elif model_type == constants.TF_KERAS:
593
- # TODO(b/141524386, b/141566408): TPU Inference is not supported
594
- # for Keras saved_model yet.
595
model = tf.keras.models.load_model(eval_saved_model_path)
596
elif model_type == constants.TF_LITE:
597
# The tf.lite.Interpreter is not thread-safe so we only load the model
0 commit comments