Revert "Trimmed the branch to only the tf2_py sub directory"

This reverts commit ee78c4a49e5fd61f23ee8152a9a53694bd73f7d7.
Rebase from 'upstream'
Trimmed the branch to only the tf2_py sub directory