You can’t pass that layout “directly”, however it should be trivial to pass the correct input representation. You would simply pass a vector that looks like:
v = [ vector, vector + N*N, vector + N*N*2, ...]
This can be done trivially with a for-loop in host code.