code: def _build_mnist_layer_list():
      return [
          l.Conv2D(
              32,
              5,
              padding="same",
              activation="relu",
              input_shape=(28, 28, 1),
          ),
          l.MaxPooling2D((2, 2), (2, 2), padding="same"),
          l.BatchNormalization(),
          l.Conv2D(64, 5, padding="same", activation="relu"),
          l.MaxPooling2D((2, 2), (2, 2), padding="same"),
          l.Flatten(),
          l.Dense(1024, activation="relu"),
          l.Dropout(0.4),
          l.Dense(10, activation="softmax"),
      ]
  
file_path: model-optimization/tensorflow_model_optimization/python/core/sparsity/keras/test_utils.py
imports: import tempfile
  import numpy
  import tensorflow
  from tensorflow_model_optimization.python.core.sparsity.keras import prune
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_utils
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_wrapper
label_desc: Use image node_id 1 for calling a global function with example usage: _build_mnist_layer_list() and returns: list