How is the patch input into the network?

This network occupies too much graphics memory. Usually, inputs of 4, 1, 128, and 128 will occupy up to 16GB of graphics memory. High definition images must be input in patches. How did the original JAX network handle this?