Your debugging so far seems reasonable and indeed the last shape might fail.
I don’t know what the solution in the other thread was besides a “dataset error”, so maybe you could also ask the author what exactly was wrong.
For the input shapes: are these shapes coming from the same tensor in different iterations? If so, which operation creates these tensors?