Remove shape_list and use actual TF functions
Oh god don't merge this one, I'm testing a codebase cleanup and seeing what breaks!
_The documentation is not available anymore as the PR was closed or merged._