code: def make_basic_ngpu(
      nb_classes=10, input_shape=(None, 28, 28, 1), **kwargs
  ):
      """
      Create a multi-GPU model similar to the basic cnn in the tutorials.
      """
      model = make_basic_cnn()
      layers = model.layers
  
      model = MLPnGPU(nb_classes, layers, input_shape)
      return model
  
file_path: cleverhans/cleverhans_v3.1.0/examples/multigpu_advtrain/make_model.py
imports: from model import Conv2D, ReLU, Flatten, Linear, Softmax, MLP
  from model import MLPnGPU
  from model import Conv2DnGPU
  from model import LinearnGPU
  from model import MaxPool
  from resnet_tf import ResNetTF
label_desc: Use image node_id 2 for calling a global function with example usage: make_basic_ngpu(nb_classes, input_shape) and returns: model