I am having a terrible time analyzing a model using Cube AI.
When I analyze the TFlite model, I get an internal error.
![Image 08-06-2024 at 11.17 PM.jpg Image 08-06-2024 at 11.17 PM.jpg](https://community.st.com/t5/image/serverpage/image-id/78675iC17F3078A33FFA6A/image-size/large/is-moderation-mode/true?v=v2&px=999)
When I analyze the same model saved in Keras, I get the error below. However, the model trains and works perfectly for inference in tensorflow.
![Image 04-06-2024 at 2.35 PM.jpg Image 04-06-2024 at 2.35 PM.jpg](https://community.st.com/t5/image/serverpage/image-id/78676i2B32E779F5C58A59/image-size/large/is-moderation-mode/true?v=v2&px=999)
I have decorated the classes, for example, this way:
@tf.keras.utils.register_keras_serializable(package="Custom", name="PatchEmbedding"), but it still doesn't work.
Can anyone help me out?