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

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