Type I_TPUReplicaContext Namespace tensorflow.python.distribute.tpu_strategy Interfaces IReplicaContext