might actually be a bug, if you can save the model and send me the URL I will check it out
https://editor.aifiddle.io/models/cloud/9f33d272-5aab-40eb-8...
Maybe the last reshape is not actually legal since it reduces the number of values?
edit: Yes, I think that's it. Using Dense instead works.