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

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