Core ML. Style transfer, maximum size of input/output

Question

I set input and output image sizes equal to 2048x2048. And have the following error in my iPhone 7:

exception=Espresso exception: "Context e