Type I_InReplicaThreadMode Namespace tensorflow.python.distribute.distribution_strategy_context Interfaces I_ThreadMode