upgrade keras related CI jobs to tf 2.8.0

[garymm]

In latest tf version, some existing functions/models have been removed from tensorflow.python.keras namespace.

Could not figure it out with one simple solution, need some extra efforts to find out a good solution.