Type I_StochasticGradientDescentHandler Namespace tensorflow.python.tpu.tpu_embedding Interfaces I_OptimizerHandler