A theoretical model for task routing in mixture-of-expert transformers