The error is raised in the model in a matrix multiplication so I guess in a linear layer.
I don’t know how the model is defined, but based on this thread you might be running into an expected shape mismatch. Check the model implementation and make sure that the tensors have the expected shapes internally. I’m not deeply familiar with the mobile backend, so I can’t exclude a valid bug in the optimization path or the backend itself.