* Revert "Refactor imports for tensorflow (#27617)"
This reverts commit 5a83a92390.
* move only some imports to top
* fix lint
* add comments
* Revert "Refactor imports for tensorflow (#27617)"
This reverts commit 5a83a92390.
* move only some imports to top
* fix lint
* add comments