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

This reverts commit bcac0cee9f76cf7041f0a95c7ac56ee472dc7dbf.
Rebase from 'upstream' (no changes)
Trimmed the branch to only the tf2_py sub directory