From 64fff76712fb30398962ecad2473a0916ee53281 Mon Sep 17 00:00:00 2001 From: Grzegorz Latosinski Date: Fri, 29 May 2026 16:32:05 +0200 Subject: [PATCH] [#97577] scripts: export_tvm_metadata: Switched method for loading Keras model --- scripts/export_tvm_metadata.py | 9 +-------- 1 file changed, 1 insertion(+), 8 deletions(-) diff --git a/scripts/export_tvm_metadata.py b/scripts/export_tvm_metadata.py index 335ee02c..ed7871f6 100644 --- a/scripts/export_tvm_metadata.py +++ b/scripts/export_tvm_metadata.py @@ -40,14 +40,7 @@ def sample_tflite_model() -> bytes: outputs = keras.layers.Dense(4, activation="softmax")(x) model = keras.Model(inputs=inputs, outputs=outputs) model.compile() - - @tf.function - def serving_fn(x): - return model(x, training=False) - - concrete_fn = serving_fn.get_concrete_function(tf.TensorSpec((1, 128, 3, 1), tf.float32)) - - return tf.lite.TFLiteConverter.from_concrete_functions([concrete_fn], model).convert() + return tf.lite.TFLiteConverter.from_keras_model(model).convert() def sample_tvm_compile(