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

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