Namespace tensorflow.python.distribute.model_collection.model_collection_base Classes model_collection_base ModelAndInput Interfaces IModelAndInput