TF CI fix for Segformer
This very small PR rewrites a couple of reshapes so the TF compiler can figure out the channels dim for Segformer even when some input dimensions are undefined. Should fix any CI issues the model has been having.
Ah, I know, I just thought you might still be unconscious!<|||||>_The documentation is not available anymore as the PR was closed or merged._