Assuming you’ve checked the input shapes and thus verified that the shape mismatch is not caused by the input, then it might indeed be a library issue (I can’t test it as I’m not familiar with OpenFL).
Could you please create an issue on GitHub so that it can be tracked and fixes, please?