Type I_SimpleModule Namespace tensorflow.python.distribute.model_collection.simple_models Interfaces IModule