In case someone stumble on this, the issue was actually with flexible input dimension, same model with fixed input shapes is behaving as expected
gsigms
Users receive reputation points by contributing quality content to the forums.
Posts include post and replies published on the forums.
Post authors can mark replies as Accepted to indicate successful solutions.
Apple Developer Forums admins can mark replies as Apple Recommended to indicate an approved solution
Post
Replies
Boosts
Views
Activity
Have you filed a bug for this?
no
Have you tested on macOS Sequoia?
the behaviour is reproducible on macOS Sequoia