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

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