Add tf and tf2 to the build