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

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