Upgrade tensorflow to 2.11 and fix broken build

[Philmod]

http://b/306150775

[Philmod]

The build is almost done and passing all tests: https://ci.kaggle.net/job/kernels/job/docker-rstats/job/fix-tensorflow/