Change type constraint of the "index" in "shape.split_at" to Shape_SizeOrIndexType

Make the type contraint consistent with other shape dialect operations.

Reviewed By: jpienaar

Differential Revision: https://reviews.llvm.org/D96377