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

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