Hi Masahi, I think dynamic split will be impossible, because the output tuple will have a unknown size. The size of a tuple must be static in Relay. However, it could be possible if the pytorch graph tells us the number of outputs and we know it to be static.
1 Like