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

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