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

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